使用ssd_mobilenet

Question

我尝试在自己的数据集中训练ssd-mobilenet： 训练图像：3400尺寸：1600 * 1200 测试集：800，大小：1600 * 1200 tensorflow -gpu：1.13.1 gpu：4GB cuda 10.0 cudnn 7 对象：道路损坏（如aligator裂缝），但经过197000步后，我的训练损失无法下降2。 我有两个问题 在使用像ssd_mobilenet这样的预训练模型之前，我应该先规范训练并设置图像吗？ 如是 我应该注释归一化的图像吗？ 我真的需要帮助。预先感谢

Answer 1

  

在使用像ssd_mobilenet这样的预训练模型之前，我应该标准化训练并设置图像吗？

不。假设您正确定义了训练管道（请参见TF模型库中的示例），那么对象检测API将负责定义所需的适当图像转换（缩放，填充，归一化等），以使输入与输入兼容。模型。