yolov5怎么不使用预训练权重

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

img

img

【以下回答由 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的要求进行组织,并将其路径配置在配置文件的trainval字段中。

  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版本中不使用预训练权重进行训练了。



【相关推荐】



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