Closed Ekta246 closed 3 years ago
Also, I am training, validating and testing (evaluating) my dataset having 1 class. In the detections, generated by the model I always see the classes == 1 in the output [x, y, x2, y2, score, class].
Can anyone throw a light if it is wrong?
https://github.com/rwightman/efficientdet-pytorch/blob/52279b43acfa1b2a653ad74c9dc8bf8ee991cbcf/effdet/anchors.py#L127
I guess the generate_detections function in anchors.py returns the shape [Max_DET_PER_IMAGE, 6] and not [MAX_DETECTION_POINTS, 6]