最近,经常碰到有人拿了网上的yolo模型进行训练,大部分数据集是不会给你的,然后拿自己的数据集训练根本跑不起。因为他们的训练脚本是和他的数据集对应的,可能为了特殊的用途,进行对应的修改。
说白了,就是没有理解对yolo模型训练的逻辑。
步骤1:是要下载到yolo官方的预训练模型。什么版本下什么的。
步骤2:把对应版本的yolo官方代码下载到本地。
步骤3:如果只是试验跑一下,就直接下载官方的数据,跑一下即可。如果需要重新根据自己的数据集训练。参照以下步骤
步骤1:使用labelimg 自己打标签,导出yolo格式的txt.如果是其他标签工具,xml要庄成txt.
步骤2:修改配置文件dataset.yaml
步骤3:
.组织训练集的目录
步骤4:选择模型backbone进行模型配置文件的修改
步骤5:训练
步骤4:如果有特殊的要求,就得根据对应数据集,修改对应训练脚本即可