yoloV5的学习-pycharm版本

真的很让人气愤的一点，老师把我的pycharm给卸载了，我那个上面不仅有gpu-torch，还有gpu-torch，他给俺删了，删了很久，我心都碎了，过几天我就去找他负责，让他给我装回来我的环境！

一、安装专业版本的pycharm 

这是我所用的环境 

二、下载yoloV5的的模型

我资源绑定失败了，应该是csdn网站上有该文件了。

三、打开yoloV5模型

四、进入环境

 cd D:\AI\yolov5-master\yolov5-master     （进入到根目录）

这块根据自己的情况来

pip install -r requirements.txt

用于根据 requirements.txt 文件安装所需的 Python 包。这个文件通常包含一个项目所需的所有依赖包的列表及其版本信息。

python .\detect.py --source .\data\images\bus.jpg    检查物品（windows命名）
python .\classify\predict.py --source .\data\images\bus.jpg
python .\segment\predict.py --source .\data\images\bus.jpg

 


结果如上，在这里找

以上是小练手，用来熟悉一下

五、下载数据集 

python classify/train.py --model yolov5s-cls.pt --data cifar10 --epochs 5 --img 224

报错1：

解决方案：

train.py文件下加上红色箭头指向的代码
报错2：

解决如下：
在源代码改一下
 

python classify/train.py --model yolov5s-cls.pt --data cifar10 --epochs 5 --img 224

注意一下这个文件的命名以及位置
以上便可以正常训练了

训练完成

2.使用训练好的模型 
 

1. 预测（Prediction）

使用训练好的模型对单张图片进行预测。

python classify/predict.py --weights runs\train-cls\exp5\weights\best.pt --source im.jpg

2. 验证（Validation）

对数据集进行验证，查看模型在验证集上的性能。
 

python classify/val.py --weights runs\train-cls\exp5\weights\best.pt --data E:\datasets\cifar10

3. 导出（Export）

将训练好的模型导出为其他格式，例如 ONNX（Open Neural Network Exchange）。
 

python export.py --weights runs\train-cls\exp5\weights\best.pt --include onnx

4. 使用 PyTorch Hub 加载模型

如果您想在其他地方使用训练好的模型，可以通过 PyTorch Hub 加载它：
 

model = torch.hub.load('ultralytics/yolov5', 'custom', 'runs\train-cls\exp5\weights\best.pt')

六、标注 
下载标注工具

pip install labelimg

|打开

labelimg 

七、使用 
 

python classify/predict.py --weights runs\train-cls\exp5\weights\best.pt --source E:\yolov5-master\images\tmp\tmp_upload.png

因为我是用data cifar10训练的模型，结果显然不是各种动物和手机没问题

baocuo
在使用中，直接调用了老师给的指令，虽然没什么问题，但是报错如下
 修改如下，大家一定要注意mask_data.yaml文件里面的内容

例如我的文件地址就不是这样的

显然我根本就没有，但是刚才cafer_10的测试基本证实模型基本能行，接下来毕设还要用，直接不做mask的数据集，直接做毕设的水果的数据集得了