Spaces:
Build error
Build error
Update Lowlight.py
Browse files- 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(
|
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)
|