if __name__ == '__main__':
"""
opt模型主要参数解析:
--weights:初始化的权重文件的路径地址
--cfg:模型yaml文件的路径地址
--data:数据yaml文件的路径地址
--hyp:超参数文件路径地址
--epochs:训练轮次
--batch-size:喂入批次文件的多少
--img-size:输入图片尺寸
--rect:是否采用矩形训练,默认False
--resume:接着打断训练上次的结果接着训练
--nosave:不保存模型,默认False
--notest:不进行test,默认False
--noautoanchor:不自动调整anchor,默认False
--evolve:是否进行超参数进化,默认False
--bucket:谷歌云盘bucket,一般不会用到

本文详细介绍了如何使用yolov5框架训练自己的目标检测模型,包括模型参数解析,如权重初始化、数据配置、训练轮次、批大小等。还提到了训练过程中的关键选项,如是否使用矩形训练、是否缓存图像以加速训练等。
最低0.47元/天 解锁文章
5561

被折叠的 条评论
为什么被折叠?



