From 97e2df124d0fa96eb7f101961a2eb806cdd25049 Mon Sep 17 00:00:00 2001 From: Peter Lin Date: Tue, 22 Dec 2020 10:47:03 -0800 Subject: [PATCH] Fix bug --- inference_images.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference_images.py b/inference_images.py index 2dcac19..07d7492 100644 --- 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)