Keiser41 commited on
Commit
0809b65
·
1 Parent(s): f199ded

Update pintar.py

Browse files
Files changed (1) hide show
  1. pintar.py +2 -2
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(img_path).convert("RGB")
78
  width, height = img1.size
79
- img2 = Image.open(ref_img_path).convert("RGB")
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)