cube_detector
cube_detector copied to clipboard
Cube Detection using YOLOv5 with Oriented Bounding Boxes (OBB)
Cube Detection with Oriented Bounding Boxes
Run inference
python detect.py --source 0 --weights m640rot.pt --imgsz 640 --conf-thres 0.75
Run as ROS node
rosrun cube_detector ros_detect.py
Training
python3 train.py --img 640 --batch 3 --epochs 3 --data aug_cube.yaml
Test
python3 detect.py --source 0 --weights m640rot.pt --img 640