bucket-yolo-obb

Bucket Handle OBB Detection

This repository contains a YOLO-based oriented bounding box (OBB) detector for detecting the handles of cylindrical buckets.

The model is designed for robotic perception and grasping applications, where the position and orientation of the bucket handles are required for subsequent grasp pose estimation.

Model Description

• Task: Oriented Bounding Box Detection
• Framework: Ultralytics YOLO
• Target Object: Bucket handles
• Application: Robot perception and grasping
• Model Format: PyTorch (.pt)
• License: AGPL-3.0

Classes

The model detects the following class:

0: handle

Modify this section if your actual class name is different.

Files

best.pt      # PyTorch model

Limitations

The model performance may decrease under conditions significantly different from the training dataset, including:

• Severe occlusion
• Extreme viewing angles
• Strong illumination changes
• Motion blur
• Bucket handles with significantly different geometry

The OBB orientation is an image-plane orientation and should not be directly interpreted as a complete 3D grasp orientation.

License

This model is released under the GNU Affero General Public License v3.0 (AGPL-3.0).