gulp>> cube_detector>> 返回
项目作者： thohemp

项目描述 ：
Cube Detection using YOLOv5 with Oriented Bounding Boxes (OBB)
高级语言： Python
项目地址: git://github.com/thohemp/cube_detector.git
创建时间： 2021-08-23T11:18:08Z
项目社区：https://github.com/thohemp/cube_detector
开源协议：
下载


Cube Detection with Oriented Bounding Boxes

Screenshot

Run inference

  1. python detect.py --source 0 --weights m640rot.pt --imgsz 640 --conf-thres 0.75

Run as ROS node

  1. rosrun cube_detector ros_detect.py

Training

  1. python3 train.py --img 640 --batch 3 --epochs 3 --data aug_cube.yaml

Test

  1. python3 detect.py --source 0 --weights m640rot.pt --img 640