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

[BUG] Wrong function description #123

Closed MichaelMonashev closed 3 years ago

MichaelMonashev commented 3 years ago

Describe the bug Here is https://github.com/rwightman/efficientdet-pytorch/blob/master/effdet/anchors.py#L128 described returned values for function generate_detections() as "each row representing [x, y, width, height, score, class]" . But function return [x_min, y_min, x_max, y_max, score, class]

rwightman commented 3 years ago

@MichaelMonashev you're welcome to submit a PR for that comment, otherwise I'll get to it with next batch of changes