如何快速上手yolov5模型的训练,以及通过训练出来的参数进行模型推理(小白教学,只教如何训练和推理,后续会就行更新如何调参)(在看本文章之前,需要保证你的环境配置成功)
(1)weights:此参数是传入模型训练时的预训练权重的,如yolov5中的yolov5s.pt文件,在本文的第二张图片中可以看到,这个是官方给的配置文件,在自己第一次训练模型时,可以拿出来使用,来提高模型的训练成效。图1中,train和val的路径如图所示放置,紧接着下面会出现classes,这个是你需要训练模型的目标的分类,如图原yolov5代码中分为80类,你可以修改成自己需要的类别,其格式和源代码保持一致。
原创
2023-05-27 15:48:48 ·
5813 阅读 ·
7 评论