AlexZou commited on
Commit
aed4629
·
1 Parent(s): bf3b4b7

Update Lowlight.py

Browse files
Files changed (1) hide show
  1. Lowlight.py +1 -1
Lowlight.py CHANGED
@@ -33,7 +33,7 @@ try:
33
  onnx_input = {'input': input_image}
34
  #onnx_output0, onnx_output1, onnx_output2 = onnx_session.run(['output0', 'output1', 'output2'], onnx_input)
35
  onnx_output = onnx_session.run(['output'], onnx_input)
36
- torchvision.utils.save_image(torch.from_numpy(onnx_ouput[0]), config.save_path+'output.png')
37
  #torch_output = np.squeeze(onnx_output[0], 0)
38
  #torch_output = np.transpose(torch_output * 255, [1, 2, 0]).astype(np.uint8)
39
  #plt.imsave(config.save_path+'output.png', torch_output)
 
33
  onnx_input = {'input': input_image}
34
  #onnx_output0, onnx_output1, onnx_output2 = onnx_session.run(['output0', 'output1', 'output2'], onnx_input)
35
  onnx_output = onnx_session.run(['output'], onnx_input)
36
+ torchvision.utils.save_image(torch.from_numpy(onnx_output[0]), config.save_path+'output.png')
37
  #torch_output = np.squeeze(onnx_output[0], 0)
38
  #torch_output = np.transpose(torch_output * 255, [1, 2, 0]).astype(np.uint8)
39
  #plt.imsave(config.save_path+'output.png', torch_output)