feat : Update code, new args

2025-08-14 09:26:37 +00:00
parent 2111d9c52c
commit 34b17b0280
7 changed files with 13 additions and 9 deletions
--- a/grounding_dino/groundingdino/models/GroundingDINO/utils.py
+++ b/grounding_dino/groundingdino/models/GroundingDINO/utils.py
@@ -79,6 +79,7 @@ def gen_encoder_output_proposals(
        grid_y, grid_x = torch.meshgrid(
            torch.linspace(0, H_ - 1, H_, dtype=torch.float32, device=memory.device),
            torch.linspace(0, W_ - 1, W_, dtype=torch.float32, device=memory.device),
+            indexing="ij"
        )
        grid = torch.cat([grid_x.unsqueeze(-1), grid_y.unsqueeze(-1)], -1)  # H_, W_, 2