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的数据集，直接做毕设的水果的数据集得了