yolov5训练流程总结
- 主要是两个脚本文件,train.py(模型训练脚本文件),detect.py(利用训练出来最好的权重文件进行推理的脚本文件)
- 流程为:先用开始模型自带的权重文件进行训练,等到模型训练完之后会生成一个最好的权重文件,然后利用这个最好的权重文件进行推理
- 需要更改的文件:一个是data目录下的相应的yaml文件,一个是model目录文件下的相应的yaml文件,一个是train.py(模型训练脚本文件)
All articles on this blog are licensed under CC BY-NC-SA 4.0 unless otherwise stated.
Comments
