Upload test_dataset.py

deepfillv2/test_dataset.py

@@ -7,10 +7,16 @@ from config import *
 
 
 class InpaintDataset(Dataset):
-    def __init__(self):
-        self.imglist = [INIMAGE]
-        self.masklist = [MASKIMAGE]
-        self.setsize = RESIZE_TO
+    def __init__(self, in_image, mask_image, resize_to):
+        if resize_to is None:
+            resize_to = RESIZE_TO
+        self.imglist = [in_image]
+        self.masklist = [mask_image]
+        self.setsize = resize_to 
+    # def __init__(self):
+    #     self.imglist = [INIMAGE]
+    #     self.masklist = [MASKIMAGE]
+    #     self.setsize = RESIZE_TO
 
     def __len__(self):
         return len(self.imglist)