Merge pull request #14 from h-bo/patch-1

fix anchor mistake
lyuwenyu · Jul 26, 2023 · 8ec57b6 · 8ec57b6
2 parents 2d22a7c + 7b2c0ef
commit 8ec57b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rtdetr_paddle/ppdet/modeling/transformers/rtdetr_transformer.py b/rtdetr_paddle/ppdet/modeling/transformers/rtdetr_transformer.py
@@ -462,7 +462,7 @@ def _generate_anchors(self,
                     end=w, dtype=dtype))
             grid_xy = paddle.stack([grid_x, grid_y], -1)
 
-            valid_WH = paddle.to_tensor([h, w]).astype(dtype)
+            valid_WH = paddle.to_tensor([w, h]).astype(dtype)
             grid_xy = (grid_xy.unsqueeze(0) + 0.5) / valid_WH
             wh = paddle.ones_like(grid_xy) * grid_size * (2.0**lvl)
             anchors.append(