| import gradio as gr | |
| def main(N, Hin, Win, kernel_size, stride, padding, dilation, output_padding): | |
| Hout = (Hin - 1 )*stride - 2*padding + dilation*(kernel_size-1) + output_padding + 1 | |
| Wout = (Win - 1 )*stride - 2*padding + dilation*(kernel_size-1) + output_padding + 1 | |
| if Hout > 0 and Wout > 0: | |
| return f"[{N}, Cout, {Hout}, {Wout}]" | |
| return "Error" | |
| iface = gr.Interface( | |
| fn=main, | |
| inputs=[*(gr.inputs.Slider(0, 2048, default=1) for i in range(3)), *[gr.inputs.Slider(0, 50, default=1, step=1) for i in range(5)]], | |
| outputs=["text"], | |
| live=True, | |
| allow_screenshot=False, | |
| allow_flagging=False) | |
| iface.launch() |