Merge PR #307 from ya0guang/patch-1

Fix the bug in `utils_motion.py`
2024-02-26 15:35:54 -06:00
parent 61481955be 5f8fba1ba3
commit 12c9879ebb
1 changed files with 1 additions and 1 deletions
@@ -149,7 +149,7 @@ def get_combined_multival(multivalA: Union[float, Tensor], multivalB: Union[floa
        areaA = multivalA.shape[1]*multivalA.shape[2]
        areaB = multivalB.shape[1]*multivalB.shape[2]
        # match height/width to mask with larger area
-        leader,follower = multivalA,multivalB if areaA >= areaB else multivalB,multivalA
+        leader,follower = (multivalA,multivalB) if areaA >= areaB else (multivalB,multivalA)
        batch_size = multivalA.shape[0] if multivalA.shape[0] >= multivalB.shape[0] else multivalB.shape[0]
        # make follower same dimensions as leader
        follower = torch.unsqueeze(follower, 1)