1. 效果:
视频链接:
最终效果:
源码已经上传 Github:
2. YOLOv5模型训练:
训练自己的数据集可以看我这篇博客:
这里演示的话我就用官方训练好的 yolov5m.pt 模型。
3. YOLOv5模型预测:
预测接口:
import torch
import numpy as np
from models.experimental import attempt_load
from utils.general import non_max_suppression, scale_coords, letterbox
from utils.torch_utils import select_device
import cv2
from random import randint
class Detector(object):
def __init__(self):
2021-09-23 14:55:53
39.76MB
Python
1