https://github.com/thohemp/cube_detector

Cube Detection using YOLOv5 with Oriented Bounding Boxes (OBB)
https://github.com/thohemp/cube_detector

computer-vision object-detection oriented-bounding-box rotation yolo yolov5

Last synced: 5 months ago
JSON representation

Cube Detection using YOLOv5 with Oriented Bounding Boxes (OBB)

Host: GitHub
URL: https://github.com/thohemp/cube_detector
Owner: thohemp
Created: 2021-08-23T11:18:08.000Z (over 3 years ago)
Default Branch: master
Last Pushed: 2022-05-25T14:24:22.000Z (almost 3 years ago)
Last Synced: 2024-12-28T08:24:53.366Z (5 months ago)
Topics: computer-vision, object-detection, oriented-bounding-box, rotation, yolo, yolov5
Language: Python
Homepage:
Size: 58 MB
Stars: 24
Watchers: 3
Forks: 9
Open Issues: 5
Metadata Files:
- Readme: README.md
- Citation: CITATION.cff

README

# Cube Detection with Oriented Bounding Boxes
![Screenshot](example_video.gif)

# Run inference
```sh
python detect.py --source 0 --weights m640rot.pt --imgsz 640 --conf-thres 0.75
```

# Run as ROS node

```sh
rosrun cube_detector ros_detect.py
```

# Training

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

# Test

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