pytorch模型部署到web(pytorch模型部署到wpf)

pytorch模型部署到web(pytorch模型部署到wpf)p style margin left 0 text align left span style background color ffffff u span style background color ffffff span u span p

大家好,我是讯享网,很高兴认识大家。



 <p style="margin-left:0;text-align:left;"><span style="background-color:#ffffff;"><u><span style="background-color:#ffffff;"><span style="color:#;"><u>取决算力资源情况&#xff0c;本着先完成再完美&#xff0c;直接win本机部署yolov5-7.0&#xff0c;边操作边梳理步骤如下&#xff1a;</u></span></span></u></span></p> 

讯享网

一、部署前提

已安装Anaconda3、PyCharm软件,若未安装建议查看教程:

Win本机安装Anaconda3和PyCharm详细教程:

https://blog.csdn.net/u0/article/details/?spm=1001.2014.3001.5502

二、项目来源

可Github获取:https://github.com/ultralytics/yolov5

三、部署过程

1、打开项目


讯享网

2、添加虚拟环境和解释器

base环境下创建名为“test”的虚拟环境

conda create -n test python==3.8.5

conda env list

3、安装依赖

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install pillow==9.5.0 -i Simple Index

请耐心等待

4、安装数据标注工具(labelimg)并运行

该图形图像标注工具标注的类型为2D矩形框

pip install labelimg==1.8.6 -i https://pypi.tuna.tsinghua.edu.cn/simple

labelimg

5、数据标注

完成剩余图像的标注

6、创建数据集目录并添加图像及其标注文件

方法1:直接到工程项目”yolov5-7.0“下创建

方法2:在PyCharm中创建【以该方法为例】

将图像拆分为:

(1)训练集train

(2)验证集val

数据规模较小时,一般按7:3拆分,70%图像及其标注文件分别复制到train中,剩余30%图像和标注文件分别复制到val中

”original_images“中共100张,编号为001-100,我只用了前10张跑了一下,并按7:3进行拆分。

【如果想效果好的话标注需要精准,图片数量也需要保证】

以训练集图像为例,截图如下

7、配置数据集文件

工程项目”yolov5-7.0“-&gt;data目录下-&gt;coco.yaml

复制一份在原路径下为coco1.yaml并打开,修改类别名称,与”5、数据标注“中所打标签一致,不确定的话可以去save时保存的路径下找到classes.txt”文件。

注意:

①类别名称、顺序需一致。

②path、train、val路径要进行替换,最好使用绝对路径,没用到test,可以注释掉。

③分类后面的download及以后内容可删除。

8、配置模型参数

工程项目”yolov5-7.0“-&gt;models目录下-&gt;yolov5s.yaml

复制一份在原路径下为yolov5s1.yaml并打开,修改第1行nc(类别数量)为 3【注意:”5、数据标注“中分几类就修改为几】

9、配置训练参数

工程项目”yolov5-7.0“-&gt;train.py

复制一份在原路径下为train1.py并打开,在下方主函数处进行参数配置,根据本机情况配置修改即可,用CPU在跑,所以将”训练迭代轮数“、”训练批量大小“都调小了。

10、启动训练

11、训练结果

去对应路径下找-&gt;result.png【流程大概就是这一趟】


小讯
上一篇 2025-05-15 10:21
下一篇 2025-04-20 16:00

相关推荐

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容,请联系我们,一经查实,本站将立刻删除。
如需转载请保留出处:https://51itzy.com/kjqy/186964.html