Spaces:
Running
Running
Update inference_img.py
Browse files- inference_img.py +0 -2
inference_img.py
CHANGED
@@ -113,8 +113,6 @@ def main():
|
|
113 |
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])
|
114 |
else:
|
115 |
cv2.imwrite('output/img{}.png'.format(i), (img_list[i][0] * 255).byte().cpu().numpy().transpose(1, 2, 0)[:h, :w])
|
116 |
-
|
117 |
-
cmd_inference_img(args.img,args.exp,args.model)
|
118 |
|
119 |
if __name__ == "__main__":
|
120 |
main()
|
|
|
113 |
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])
|
114 |
else:
|
115 |
cv2.imwrite('output/img{}.png'.format(i), (img_list[i][0] * 255).byte().cpu().numpy().transpose(1, 2, 0)[:h, :w])
|
|
|
|
|
116 |
|
117 |
if __name__ == "__main__":
|
118 |
main()
|