Fix the bug in utils_motion.py

This statement is syntactically correct, but its behavior might not match the expected outcome due to the way Python's conditional (ternary) operator precedence works. In Python, the conditional expression has lower precedence than the comma.
2024-02-26 15:58:14 -05:00
parent 61481955be
commit 5f8fba1ba3
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)