From d7b4600ad679d9e2e4a7dda023cf517b4a5ec73e Mon Sep 17 00:00:00 2001 From: zbhuang Date: Sun, 18 Feb 2024 19:26:11 -0500 Subject: [PATCH] impl: set indexing=ij to avoid warning --- ptlflow/models/flowformer/utils.py | 1 + ptlflow/models/flowformerplusplus/utils.py | 1 + ptlflow/models/matchflow/utils.py | 1 + 3 files changed, 3 insertions(+) diff --git a/ptlflow/models/flowformer/utils.py b/ptlflow/models/flowformer/utils.py index ab0b1d0..47c5e7b 100644 --- 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 index 4c54eea..183a184 100644 --- 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 index 23b6367..c854cc0 100644 --- 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