Чтобы обучить новую модель, обучающая запись: main.py .
Вот несколько шагов, как добавить новые модели.
Если вы хотите Чтобы просто построить простую модель, скажем, MyRetin aNet, поверх существующих существующих компонентов, таких как слои, потери, существующие головки, вам может потребоваться:
- Добавить шаблон конфигурации для новой модели, например this .
- Добавьте файл "my_retinanet_model.py" в папку моделирования (аналогично "retinanet_model.py") и внедрите модель.
- Добавьте ветвь к заводскому файлу , чтобы вы могли использовать его в main.py .
Если вы хотите добавить некоторые мелкозернистые компоненты, такие как головки и магистрали , затем вам нужно что-то добавить в папку models / official / vision / discovery / моделирование / Architecture / Architecture / .
- Добавить класс в heads.py (для головы) или новый .py файл для магистралей.
- Обновите factory.py соответственно.
- Возможно, вам также понадобится d для изменения конфигурации модели соответственно,
Для более мелких операций можно добавить к ops и utils .