simple yolov5 wrapper for detect(onnx, dnn, openvino)
Project description
YOLOv5 detect
🚀simple yolov5 wrapper for detect(onnx, dnn, openvino)
Installation
pip install yolov5-detect
Example
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
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
yolov5-detect-0.1.21.tar.gz
(77.6 kB
view hashes)
Built Distribution
yolov5_detect-0.1.21-py3-none-any.whl
(112.8 kB
view hashes)
Close
Hashes for yolov5_detect-0.1.21-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | f6107d7ea6d509623382e0e2eb4ee5ef142abc5fd8946d544efaa338999ce056 |
|
MD5 | 0db54ebb3f6e34d2ee26c021fb4250a6 |
|
BLAKE2b-256 | efa099f0b3e1694f18ecebc2e3d946e0976783a407ab2e8b4038dbca8434ccea |