训练自己的数据集，并使用训练好的模型进行测试（少走弯路）

[yfq512]

• 1. 准备数据，制作训练时需要的文件（data文件夹下面的四个文件：数据类名，锚框，训练数据，测试数据）
• 2. 根据步骤1修改core/config.py文件
• 3. 基于yolo原始权重训练，（或者从头训练）（这步教程写的很清楚）
• 4. 当loss为1左右时，模型就已经有效果了，训练过程中会保存很多模型，选择一个loss最小的，并根据所选择的模型名修改freeze_graph.py，修改后运行freeze_graph.py，会在项目根目录生成***.pb文件
• 5. 修改demo文件，如修改video_demo.py，修改调用的***.pb文件路径，修改num_class为自己数据类个数

[Jugerr]

请问训练自己的数据需要把图片缩放到一样的大小吗？

[XieQiang111]

谢谢，很有帮助

[laisimiao]

@yfq512 你好。在iv步的时候，不需要先convert_weights.py运行然后再运行freeze_graph.py吗？,因为demo是这样运行的？

[nipperhou]

你好，请问我准备使用该网络进行单目标检测，检测出的test loss 保持在40左右，然后检测出的框完全不对，都框在图片的边缘，请问你知道是哪方面的问题吗？（数据集总共有355张）

[XieQiang111]

可能的问题有： 1.annotations文件有问题 2.训练数据太少 3.test loss太大，训练没收敛

------------------ 原始邮件 ------------------ 发件人: "YunYang1994/tensorflow-yolov3" @.>; 发送时间: 2021年3月26日(星期五) 下午4:12 @.>; @.@.>; 主题: Re: [YunYang1994/tensorflow-yolov3] 训练自己的数据集，并使用训练好的模型进行测试（少走弯路） (#588)

你好，请问我准备使用该网络进行单目标检测，检测出的test loss 保持在40左右，然后检测出的框完全不对，都框在图片的边缘，请问你知道是哪方面的问题吗？（数据集总共有355张）

— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.

[nipperhou]

感谢

[StarStar-666]

怎么把模型参数(.ckpt文件)回到最原始的yolov3大小?

[DragMou1204]

请问，训练完成生成的文件名loss显示nan可能是什么错误呢，我训练的是自己的数据集