MCPcopy Create free account
hub / github.com/drinkingcoder/FlowFormer-Official / __call__

Method __call__

core/utils/augmentor.py:274–284  ·  view source on GitHub ↗
(self, img1, img2, flow, valid)

Source from the content-addressed store, hash-verified

272
273
274 def __call__(self, img1, img2, flow, valid):
275 img1, img2 = self.color_transform(img1, img2)
276 img1, img2 = self.eraser_transform(img1, img2)
277 img1, img2, flow, valid = self.spatial_transform(img1, img2, flow, valid)
278
279 img1 = np.ascontiguousarray(img1)
280 img2 = np.ascontiguousarray(img2)
281 flow = np.ascontiguousarray(flow)
282 valid = np.ascontiguousarray(valid)
283
284 return img1, img2, flow, valid

Callers

nothing calls this directly

Calls 3

color_transformMethod · 0.95
eraser_transformMethod · 0.95
spatial_transformMethod · 0.95

Tested by

no test coverage detected