Fix bug

PyWarnFixerRepos · Dec 22, 2020 · 97e2df1 · 97e2df1
1 parent 55d0ddc
commit 97e2df1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference_images.py b/inference_images.py
@@ -95,7 +95,7 @@
     ImagesDataset(args.images_bgr),
 ], assert_equal_length=True, transforms=A.PairCompose([
     HomographicAlignment() if args.preprocess_alignment else A.PairApply(nn.Identity()),
-    A.PairApply(T.ToTensor)
+    A.PairApply(T.ToTensor())
 ]))
 dataloader = DataLoader(dataset, batch_size=1, num_workers=8, pin_memory=True)