search

rwightman / efficientdet-pytorch

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
Apache License 2.0
1.58k stars 293 forks source link

Shape of the detections #150

Closed Ekta246 closed 3 years ago

Ekta246 commented 3 years ago

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]

Ekta246 commented 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?