Skip to content

Commit 040cf5a

Browse files
authored
Update inference.py
1 parent fe1aede commit 040cf5a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

inference.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,15 +35,15 @@ def generate_data(opt, im_path, cloth_path, pose_path, segm_path):
3535

3636
c = Image.open(cloth_path)
3737
#resizing into size of model's input
38-
c = c.resize((opt.fine_width, opt.fine_height))
38+
c = c.resize((opt.fine_width, opt.fine_height))
3939
c = transform(c)
4040

4141
im = Image.open(osp.join(im_path))
42-
im = im.resize((opt.fine_width, opt.fine_height))
42+
im = im.resize((opt.fine_width, opt.fine_height))
4343
im = transform(im)
4444

4545
im_parse = Image.open(segm_path)
46-
im_parse = im_parse.resize((opt.fine_width, opt.fine_height))
46+
im_parse = im_parse.resize((opt.fine_width, opt.fine_height))
4747
parse_array = np.array(im_parse)
4848

4949
parse_shape = (parse_array > 0).astype(np.float32)

0 commit comments

Comments
 (0)