Skip to content

Cube Detection using YOLOv5 with Oriented Bounding Boxes (OBB)

Notifications You must be signed in to change notification settings

thohemp/cube_detector

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Cube Detection with Oriented Bounding Boxes

Screenshot

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