Wheat Detection
Overview
Репозиторий содержит код для обучения и инференса детектора Faster-RCNN в задаче детектирования колосьев пшеницы.
Run
Запуск всего пайплайна происходит с помощью run.py
. Для обучения детектора необходимо указать в config.py
параметр mode="train"
, для инференса mode="inference"
и также img_path
.