Error in detection and mAP calculation on custom dataset

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite

https://github.com/hunglc007/tensorflow-yolov4-tflite

MIT License

2.23k stars 1.24k forks source link

Error in detection and mAP calculation on custom dataset #454

Open poojatambe opened 2 years ago

poojatambe commented 2 years ago

I have trained the yolov4 from scratch using my custom dataset and completed the training on Tensorflow. Now, I am stuck at evaulation part. Could you please help on how to evaluate performance after custom dataset training?I tried using the checkpoints folder but I am not sure which file is to be used for detection. How to calculate mAP?