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