Peter Lin 4 年之前
父節點
當前提交
97e2df124d
共有 1 個文件被更改,包括 1 次插入1 次删除
  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)