【文件属性】:
文件名称:Yolov5-Flask-VUE:基于Flask开发后端、VUE开发前端框架,在WEB端部署YOLOv5目标检测模型
文件大小:39.76MB
文件格式:ZIP
更新时间:2021-05-07 11:29:14
Python
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):
【文件预览】:
Yolov5-Flask-VUE-master
----back-end()
--------core()
--------data()
--------models()
--------static()
--------utils()
--------processor()
--------uploads()
--------weights()
--------app.py(3KB)
----LICENSE(1KB)
----front-end()
--------.gitignore(231B)
--------package.json(1KB)
--------package-lock.json(515KB)
--------src()
--------public()
--------vue.config.js(214B)
--------babel.config.js(53B)
----README.md(9KB)