BrickCraft model checkpoints

Task-specific YOLOv8n-seg checkpoints used by the single-arm BrickCraft assembly pipeline.

• Project website
• Paper

Included checkpoints

Path	Purpose	Class IDs
yolo/seg_structure.pt	Segment the assembled brick structure for scene alignment	0: object
yolo/seg_brick_gripper.pt	Segment the target brick and gripper	0: brick, 1: gripper

See the YOLO checkpoint notes for loading instructions and provenance. The checkpoint metadata records Ultralytics version 8.4.8; packaging did not include a compatibility test with other versions.

Download

hf download JerryYuu/BrickCraft --local-dir ./models/BrickCraft

From the BrickCraft code repository root, configure:

{
  "yolo_seg_structure_ckpt": "./models/BrickCraft/yolo/seg_structure.pt",
  "yolo_seg_brick_gripper_ckpt": "./models/BrickCraft/yolo/seg_brick_gripper.pt"
}

For reproducibility, pin --revision to the model repository commit used in your experiment.

Scope and verification

This release contains the two YOLO checkpoints only. SAM-2 and Brick Diffusion policy checkpoints are not included. These files alone do not provide the complete robot execution stack.

The checkpoint files are preserved without modification. Their ZIP archive integrity and SHA256 checksums were verified during packaging. Inference accuracy, CUDA execution, and physical robot operation were not tested during packaging.

License

The included YOLO checkpoints are based on Ultralytics YOLOv8 and are distributed under AGPL-3.0. See LICENSE and Ultralytics licensing information. This statement concerns the files included in this model repository; additional models added in future releases must document their own applicable terms.