yolov7-package

Bindings for yolov7 in one class


Keywords
pytorch, yolo, yolov7, yolov7-tiny
License
MIT
Install
pip install yolov7-package==0.0.12

Documentation

GitHub PyPI version PyPI - Downloads

This package is deprecated. Now use DetExecutor

WongKinYiu/yolov7 as independent package

Bindings for yolov7 project (https://github.com/WongKinYiu/yolov7)

Example of usage:

from yolov7_package import Yolov7Detector
import cv2

if __name__ == '__main__':
    img = cv2.imread('img.jpg')
    det = Yolov7Detector(traced=False)
    classes, boxes, scores = det.detect(img)
    img = det.draw_on_image(img, boxes[0], scores[0], classes[0])

    cv2.imshow("image", img)
    cv2.waitKey()

You can use traced=True option to download and infer traced FP16 version of yolov7 model!

PYPI link: https://pypi.org/project/yolov7-package/