yolov5怎么不使用预训练权重

Question

yolov5 6.0版本不使用预训练权重报错
我之前使用6.2版本的将weights的yolov5s.pt删除训练
但是6.0出错了怎么办

Answer 1

【以下回答由 GPT 生成】

答案:

在yolov5的6.0版本中，可以通过以下步骤来不使用预训练权重进行训练：

1. 首先，确保您已经安装了yolov5和相关的依赖项。可以通过以下命令进行安装：

pip install -U -r requirements.txt

1. 接下来，创建一个新的训练配置文件，如custom_yolov5s.yaml。

2. 在配置文件中，将weights: yolov5s.pt这一行注释掉，或者将其更改为一个空字符串weights: ''。

3. 修改配置文件中的nc参数，将其设置为您自己的数据集中的类别数量。

4. 如果您需要更改模型的输入分辨率，请修改配置文件中的img_size参数。

5. 确保您的数据集按照yolov5的要求进行组织，并将其路径配置在配置文件的train和val字段中。

6. 然后，使用以下命令来启动训练：

python train.py --batch <batch_size> --cfg custom_yolov5s.yaml --data <path_to_data_config_file> --img <image_size>

请确保用实际的值替换<batch_size>，<path_to_data_config_file>和<image_size>。

这样，您就可以在yolov5的6.0版本中不使用预训练权重进行训练了。

【相关推荐】

• 帮你找了个相似的问题, 你可以看下: https://ask.csdn.net/questions/7688652
• 这篇博客也不错, 你可以看下yolov3用训练过得weights文件继续训练
• 您还可以看一下 卢未来老师的搞定系列:yolov5检测封装成类调用课程中的 认识yolov5框架小节, 巩固相关知识点
  
• 除此之外, 这篇博客: yolov5-v3.0尝鲜中的 下载weights 部分也许能够解决你的问题。

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^