自定义数据集上的EfficientDet mAP评估



我正在尝试运行"mAP_evaluation.py"以在我自己的数据集上获得mAP评估:https://github.com/Tessellate-Imaging/Monk_Object_Detection/tree/master/4_efficientdet/lib

但我认为整个python文件只是为COCO数据集创建的,但如果我使用函数evaluatate_COCO((,那么我不知道如何自定义我的数据集以匹配该函数。请帮忙。

p/S:我已经训练并导出了EfficientDet模型(pth文件(,预测了测试图像/视频,只是不知道如何评估。

您可以修复类似的问题

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)

相关内容

  • 没有找到相关文章

最新更新