code de inference yolov5
Project description
Package dung de inference model yolov5 voi custom object
Install
pip install yolov5-inference==0.6
Use
from yolov5_inference.infer import DetectObject
class_name = {0: 'text', 1: 'title', 2: 'list', 3: 'table', 4: 'figure'}
model = DetectObject(weights='../path/of/weights', classes=class_name)
img_path = '../path/of/img'
rs = model.detect(img_path)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
yolov5_inference-0.7.tar.gz
(63.3 kB
view hashes)