上次做了一个皮卡丘识别的模型，如下图，不知道怎么搭建yolov5训练模型，这次就是把这些训练的模型整合起来，将yolov5模型部署到web端，实现动态选择模型，上传或直接拖拽图片至窗口就能返回识别结果与json格式文件

演示：

资源下载路径👇

yolov5模型部署到web端-深度学习文档类资源

先到文件的路径下执行pip install -r requirements.txt 更新一下包

python server.py启动

 打开http://localhost:8000/

在model_selection_options这里添加模型选项，这是我提前在网上找了一些图片，训练出来的货车运输识别模型。

C: \Users\你的用户名\.cache\torch\ultralytics_yolov5_master

预训练权重：torch.hub.load('ultralytics/yolov5', model_name, pretrained=True)

而我的话是使用了自定义模型，使用本地的模型，先定义一个path，给path赋值相对路径，model_name是我们上面步骤模型选项的结果[光缆运输或者是木头运输]，source=local就是使用本地路径。具体可参考 torch.hub — PyTorch 1.13 文档

 自定义权重：torch.hub.load(path,model_name, source='local')

添加完之后去到上图设置的path = 'torch/hub/shlhc_bch_master'路径中找到hubconf.py，为刚刚的模型选项配置对应的权重文件

 我这里权重文件都统一放在weight文件夹中，所以配置信息都是'weight/****'