如何使用模型/官方/视觉/检测训练用于对象检测的自定义模型？

Question

如何使用models/official/vision/detection训练用于对象检测的自定义模型？

Answer 1

要训练新模型，训练条目为main.py。

以下是添加新模型的几个步骤。

如果您只想建立一个简单的模型（例如MyRetinaNet），则需要在现有的现有组件（例如层，损耗，现有磁头）之上，

1. 为新模型添加配置模板，例如this one。
2. 在建模文件夹中添加文件“ my_retinanet_model.py”（类似于“ retinanet_model.py”）并实现模型。
3. 在factory file中添加分支，以便您可以在main.py中使用它。

如果要添加一些细粒度的组件（例如头和骨架），则需要在models/official/vision/detection/modeling/architecture/文件夹中添加一些内容。

1. 将一个类添加到heads.py（用于heads）或一个新的.py文件用于主干。
2. 相应地更新factory.py。
3. 您可能还需要相应地更改the model configs，

对于更细粒度的操作，您可以添加到ops和utils。