Spaces:
Sleeping
Sleeping
narugo1992
commited on
Commit
•
f4177d4
1
Parent(s):
726a752
dev(narugo): add new model of caihongmao
Browse files
app.py
CHANGED
@@ -98,6 +98,7 @@ def image_to_tensor(pic: Image.Image, size: int = 512, keep_ratio: float = True,
|
|
98 |
|
99 |
|
100 |
MODELS = [
|
|
|
101 |
'TResnet-D-FLq_ema_6-30000.onnx',
|
102 |
'TResnet-D-FLq_ema_6-10000.onnx',
|
103 |
'TResnet-D-FLq_ema_4-10000.onnx',
|
@@ -180,7 +181,8 @@ if __name__ == '__main__':
|
|
180 |
gr_input_image = gr.Image(type='pil', label='Original Image')
|
181 |
with gr.Row():
|
182 |
gr_threshold = gr.Slider(0.0, 1.0, 0.7, label='Tagging Confidence Threshold')
|
183 |
-
gr_image_size = gr.Slider(128, 960, 640, step=32, label='Image for Recognition')
|
|
|
184 |
gr_keep_ratio = gr.Checkbox(value=False, label='Keep the Ratio')
|
185 |
with gr.Row():
|
186 |
gr_model = gr.Dropdown(MODELS, value=DEFAULT_MODEL, label='Model')
|
|
|
98 |
|
99 |
|
100 |
MODELS = [
|
101 |
+
'caformer_m36-3-80000.onnx',
|
102 |
'TResnet-D-FLq_ema_6-30000.onnx',
|
103 |
'TResnet-D-FLq_ema_6-10000.onnx',
|
104 |
'TResnet-D-FLq_ema_4-10000.onnx',
|
|
|
181 |
gr_input_image = gr.Image(type='pil', label='Original Image')
|
182 |
with gr.Row():
|
183 |
gr_threshold = gr.Slider(0.0, 1.0, 0.7, label='Tagging Confidence Threshold')
|
184 |
+
# gr_image_size = gr.Slider(128, 960, 640, step=32, label='Image for Recognition')
|
185 |
+
gr_image_size = gr.Slider(128, 960, 448, step=32, label='Image for Recognition')
|
186 |
gr_keep_ratio = gr.Checkbox(value=False, label='Keep the Ratio')
|
187 |
with gr.Row():
|
188 |
gr_model = gr.Dropdown(MODELS, value=DEFAULT_MODEL, label='Model')
|