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)
|