Блог

Главная - Вопросы по программированию - Оценка карты EffiecientDet для пользовательского набора данных

Оценка карты EffiecientDet для пользовательского набора данных

#python #object-detection #evaluation #efficientnet

#python #обнаружение объекта #оценка #эффективная сеть

Вопрос:

Я пытаюсь запустить ‘mAP_evaluation.py’ чтобы получить оценку карты для моего собственного набора данных: https://github.com/Tessellate-Imaging/Monk_Object_Detection/tree/master/4_efficientdet/lib

но я думаю, что весь файл python создан только для набора данных COCO, но если я использую функцию evaluate_coco(), то я не знаю, как настроить свой набор данных в соответствии с функцией. Пожалуйста, помогите.

P / S: я уже обучил и экспортировал модель EfficientDet (файл pth), прогнозируемые тестовые изображения / видео, просто не знаю, как оценить.

Ответ №1:

вы можете исправить проблему следующим образом

 def __init__(root_dir, img_dir='images', set_dir='train2017', transform=None)

итак, я исправил прямо здесь из mAP_evaluation.py:

 dataset_val = CocoDataset("/content/Monk_Object_Detection/4_efficientdet/lib/data/pothole", img_dir='images', set_dir='val2017',
                            transform=transforms.Compose([Normalizer(), Resizer()]))

evaluate_coco(dataset_val, efficientdet)