Fediory commited on
Commit
c9dce21
·
1 Parent(s): 8ced22f

fix; description

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -64,15 +64,15 @@ interface = gr.Interface(
64
  fn=process_image,
65
  inputs=[
66
  gr.Image(label="Low-light Image", type="pil"),
67
- gr.Radio(choices=['Yes','No'],label="Image Score",info="Calculation NIQE and BRISQUE, default is \"No\"."),
68
  gr.Radio(choices=pth_files2,label="Model Weights",info="Choose your model. The best models are \"SICE.pth\" and \"generalization.pth\"."),
69
- gr.Slider(0.1,10,label="gamma curve",step=0.01,value=1.0, info="Best range is [0.5,2.5]."),
70
  gr.Slider(0,2,label="Alpha-s",step=0.01,value=1.0, info="Higher is more saturated."),
71
- gr.Slider(0.1,2,label="Alpha-i",step=0.01,value=1.0, info="Higher is more lighted.")
72
  ],
73
  outputs=[
74
  gr.Image(label="Result", type="pil"),
75
- gr.Textbox(label="NIQE")
76
  ],
77
  title="HVI-CIDNet (Low-Light Image Enhancement)",
78
  description="The demo of paper \"You Only Need One Color Space: An Efficient Network for Low-light Image Enhancement\"",
 
64
  fn=process_image,
65
  inputs=[
66
  gr.Image(label="Low-light Image", type="pil"),
67
+ gr.Radio(choices=['Yes','No'],label="Image Score",info="Calculate NIQE, default is \"No\"."),
68
  gr.Radio(choices=pth_files2,label="Model Weights",info="Choose your model. The best models are \"SICE.pth\" and \"generalization.pth\"."),
69
+ gr.Slider(0.1,5,label="gamma curve",step=0.01,value=1.0, info="Lower is lighter, best range is [0.5,2.5]."),
70
  gr.Slider(0,2,label="Alpha-s",step=0.01,value=1.0, info="Higher is more saturated."),
71
+ gr.Slider(0.1,2,label="Alpha-i",step=0.01,value=1.0, info="Higher is lighter.")
72
  ],
73
  outputs=[
74
  gr.Image(label="Result", type="pil"),
75
+ gr.Textbox(label="NIQE",info="Lower is better.")
76
  ],
77
  title="HVI-CIDNet (Low-Light Image Enhancement)",
78
  description="The demo of paper \"You Only Need One Color Space: An Efficient Network for Low-light Image Enhancement\"",