0
点赞
收藏
分享

微信扫一扫

yolov5模型部署到web端,识别返回json格式文件


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

yolov5模型部署到web端,识别返回json格式文件_bc

演示:

资源下载路径👇

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

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

yolov5模型部署到web端,识别返回json格式文件_web端_02

 然后执行python server.py启动

yolov5模型部署到web端,识别返回json格式文件_权重_03

http://localhost:8000/

编辑  编辑

编辑

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

yolov5模型部署到web端,识别返回json格式文件_bc_04

yolov5模型部署到web端,识别返回json格式文件_web端_05编辑

yolov5模型部署到web端,识别返回json格式文件_权重_06

 在这里我修改了一个参数,如下它会去hub下载预训练权重,默认会下到 C: \Users\你的用户名\.cache\torch\ultralytics_yolov5_master

yolov5模型部署到web端,识别返回json格式文件_权重_07

预训练权重: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')

yolov5模型部署到web端,识别返回json格式文件_权重_08

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

yolov5模型部署到web端,识别返回json格式文件_web端_09

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

yolov5模型部署到web端,识别返回json格式文件_web端_10



举报

相关推荐

0 条评论