Update pintar.py
Browse files
pintar.py
CHANGED
@@ -74,9 +74,9 @@ if __name__ == "__main__":
|
|
74 |
img_path = args.input_folder
|
75 |
ref_img_path = args.reference_image
|
76 |
|
77 |
-
img1 = Image.open(
|
78 |
width, height = img1.size
|
79 |
-
img2 = Image.open(
|
80 |
|
81 |
img1, img1_lab = preprocessing(img1)
|
82 |
img2, img2_lab = preprocessing(img2)
|
|
|
74 |
img_path = args.input_folder
|
75 |
ref_img_path = args.reference_image
|
76 |
|
77 |
+
img1 = Image.open(ref_img_path).convert("RGB")
|
78 |
width, height = img1.size
|
79 |
+
img2 = Image.open(img_path).convert("RGB")
|
80 |
|
81 |
img1, img1_lab = preprocessing(img1)
|
82 |
img2, img2_lab = preprocessing(img2)
|