Peter Lin před 4 roky
rodič
revize
97e2df124d
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  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)