Peter Lin 4 tahun lalu
induk
melakukan
97e2df124d
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      inference_images.py

+ 1 - 1
inference_images.py

@@ -95,7 +95,7 @@ dataset = ZipDataset([
     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)