https://github.com/2432001677/yolov5-detect
simple yolov5 wrapper for detect(onnx, dnn, openvino)
https://github.com/2432001677/yolov5-detect
deep-learning detection opencv opencv-python openvino python yolov5
Last synced: 12 months ago
JSON representation
simple yolov5 wrapper for detect(onnx, dnn, openvino)
- Host: GitHub
- URL: https://github.com/2432001677/yolov5-detect
- Owner: BruceCatYu
- License: gpl-3.0
- Created: 2022-02-23T06:07:44.000Z (over 4 years ago)
- Default Branch: master
- Last Pushed: 2024-03-08T08:04:45.000Z (over 2 years ago)
- Last Synced: 2025-06-09T06:54:09.599Z (about 1 year ago)
- Topics: deep-learning, detection, opencv, opencv-python, openvino, python, yolov5
- Language: Python
- Homepage:
- Size: 447 KB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# YOLOv5 detect
🚀simple yolov5 wrapper for detect(onnx, dnn, openvino)
## Installation
```bash
pip install yolov5-detect
```
## Example
```python
import cv2
from pathlib import Path
from yolov5detect import detect, annotation
output = Path("output dir")
if not output.exists():
output.mkdir()
yolo = detect.YoloDetect("test.onnx", "test.yaml", 1000)
names = yolo.get_names()
root = Path("images dir")
for item in root.rglob("*.jpg"): # detect all images and save label results
img = cv2.imread(str(item))
det = yolo.detect(img.copy(), 0.4)
if len(det) > 0:
annotator = annotation.Annotator(img.copy())
for *xyxy, conf, cls in det:
annotator.box_label(xyxy, f"{names[int(cls)]} {conf:.2f}")
cv2.imwrite(annotator.result(), str(output / item.name))
```
## Reference
* https://github.com/ultralytics/yolov5
* https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch