__call__ -> forward()

Author: NicolasHug

references/optical_flow/presets.py

@@ -15,7 +15,7 @@ def __init__(self):
             ]
         )
 
-    def __call__(self, img1, img2, flow, valid):
+    def forward(self, img1, img2, flow, valid):
         return self.transforms(img1, img2, flow, valid)
 
 
@@ -60,5 +60,5 @@ def __init__(
         ]
         self.transforms = T.Compose(transforms)
 
-    def __call__(self, img1, img2, flow, valid):
+    def forward(self, img1, img2, flow, valid):
         return self.transforms(img1, img2, flow, valid)

references/optical_flow/transforms.py

@@ -249,7 +249,7 @@ class Compose:
     def __init__(self, transforms):
         self.transforms = transforms
 
-    def __call__(self, img1, img2, flow, valid_flow_mask):
+    def forward(self, img1, img2, flow, valid_flow_mask):
         for t in self.transforms:
             img1, img2, flow, valid_flow_mask = t(img1, img2, flow, valid_flow_mask)
         return img1, img2, flow, valid_flow_mask