tensorflow-model-garden - 如何使用模型/官方/视觉/检测训练自定义模型进行对象检测？

Question

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

Answer 1

要训​​练一个新模型，训练入口是main.py。

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

如果您只想构建一个简单的模型，比如 MyRetinaNet，在当前现有组件（如层、损失、现有头）之上，您可能需要：

1. 像这样为新模型添加一个配置模板。
2. 在建模文件夹中添加一个文件“my_retinanet_model.py”（类似于“retinanet_model.py”）并实现模型。
3. 向工厂文件添加一个分支，以便您可以在main.py中使用它。

如果你想添加一些细粒度的组件，比如头部和主干，那么你需要在models/official/vision/detection/modeling/architecture/文件夹中添加一些东西。

1. 将一个类添加到 head.py（用于头部）或一个新的 .py 文件用于主干。
2. 相应地更新factory.py。
3. 您可能还需要相应地更改模型配置，

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