#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)