impl: set indexing=ij to avoid warning

hmorimitsu · Feb 19, 2024 · d7b4600 · d7b4600
1 parent d593f06
commit d7b4600
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/ptlflow/models/flowformer/utils.py b/ptlflow/models/flowformer/utils.py
@@ -141,6 +141,7 @@ def compute_weight(hws, image_shape, patch_size, sigma=1.0, wtype="gaussian", de
     h, w = torch.meshgrid(
         torch.arange(patch_size[0], device=device),
         torch.arange(patch_size[1], device=device),
+        indexing='ij',
     )
     h, w = h / float(patch_size[0]), w / float(patch_size[1])
     c_h, c_w = 0.5, 0.5

diff --git a/ptlflow/models/flowformerplusplus/utils.py b/ptlflow/models/flowformerplusplus/utils.py
@@ -140,6 +140,7 @@ def compute_weight(hws, image_shape, patch_size, sigma=1.0, wtype="gaussian", de
     h, w = torch.meshgrid(
         torch.arange(patch_size[0], device=device),
         torch.arange(patch_size[1], device=device),
+        indexing='ij',
     )
     h, w = h / float(patch_size[0]), w / float(patch_size[1])
     c_h, c_w = 0.5, 0.5

diff --git a/ptlflow/models/matchflow/utils.py b/ptlflow/models/matchflow/utils.py
@@ -268,6 +268,7 @@ def compute_weight(hws, image_shape, patch_size, sigma=1.0, wtype="gaussian", de
     h, w = torch.meshgrid(
         torch.arange(patch_size[0], device=device),
         torch.arange(patch_size[1], device=device),
+        indexing='ij',
     )
     h, w = h / float(patch_size[0]), w / float(patch_size[1])
     c_h, c_w = 0.5, 0.5