我正在尝试运行"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)