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

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 中使用它.

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

1. 将一个类添加到 heads.py(用于头部)或一个新的 .py 文件用于 Backbone 。
2. 更新factory.py相应地。
3. 您可能还需要更改 the model configs因此，

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