Wheat Detection

Overview

Репозиторий содержит код для обучения и инференса детектора Faster-RCNN в задаче детектирования колосьев пшеницы.

Run

Запуск всего пайплайна происходит с помощью run.py. Для обучения детектора необходимо указать в config.py параметр mode="train", для инференса mode="inference" и также img_path.