AItool commited on
Commit
f55b9dc
·
verified ·
1 Parent(s): bbda141

Update inference_img.py

Browse files
Files changed (1) hide show
  1. inference_img.py +2 -2
inference_img.py CHANGED
@@ -99,8 +99,8 @@ import warnings
99
  os.mkdir('output')
100
  for i in range(len(img_list)):
101
  if args.img[0].endswith('.exr') and args.img[1].endswith('.exr'):
102
- cv2.imwrite('/output/img{}.exr'.format(i), (img_list[i][0]).cpu().numpy().transpose(1, 2, 0)[:h, :w], [cv2.IMWRITE_EXR_TYPE, cv2.IMWRITE_EXR_TYPE_HALF])
103
  else:
104
- cv2.imwrite('/output/img{}.png'.format(i), (img_list[i][0] * 255).byte().cpu().numpy().transpose(1, 2, 0)[:h, :w])
105
 
106
 
 
99
  os.mkdir('output')
100
  for i in range(len(img_list)):
101
  if args.img[0].endswith('.exr') and args.img[1].endswith('.exr'):
102
+ cv2.imwrite('output/img{}.exr'.format(i), (img_list[i][0]).cpu().numpy().transpose(1, 2, 0)[:h, :w], [cv2.IMWRITE_EXR_TYPE, cv2.IMWRITE_EXR_TYPE_HALF])
103
  else:
104
+ cv2.imwrite('output/img{}.png'.format(i), (img_list[i][0] * 255).byte().cpu().numpy().transpose(1, 2, 0)[:h, :w])
105
 
106