Spaces:
Sleeping
Sleeping
Daniel Cerda Escobar
commited on
Commit
·
53b109e
1
Parent(s):
ca59b6a
Update files
Browse files
app.py
CHANGED
@@ -92,21 +92,21 @@ with col2:
|
|
92 |
with col3:
|
93 |
st.markdown('##### Set model parameters')
|
94 |
slice_size = st.slider(
|
95 |
-
label = '
|
96 |
min_value=256,
|
97 |
max_value=1024,
|
98 |
value=768,
|
99 |
step=256
|
100 |
)
|
101 |
overlap_ratio = st.slider(
|
102 |
-
label = '
|
103 |
min_value=0.0,
|
104 |
max_value=0.5,
|
105 |
value=0.1,
|
106 |
step=0.1
|
107 |
)
|
108 |
postprocess_match_threshold = st.slider(
|
109 |
-
label = '
|
110 |
min_value = 0.0,
|
111 |
max_value = 1.0,
|
112 |
value = 0.75,
|
|
|
92 |
with col3:
|
93 |
st.markdown('##### Set model parameters')
|
94 |
slice_size = st.slider(
|
95 |
+
label = 'Slice Size',
|
96 |
min_value=256,
|
97 |
max_value=1024,
|
98 |
value=768,
|
99 |
step=256
|
100 |
)
|
101 |
overlap_ratio = st.slider(
|
102 |
+
label = 'Overlap Ratio',
|
103 |
min_value=0.0,
|
104 |
max_value=0.5,
|
105 |
value=0.1,
|
106 |
step=0.1
|
107 |
)
|
108 |
postprocess_match_threshold = st.slider(
|
109 |
+
label = 'Confidence Threshold',
|
110 |
min_value = 0.0,
|
111 |
max_value = 1.0,
|
112 |
value = 0.75,
|
utils.py
CHANGED
@@ -30,7 +30,7 @@ def sahi_yolov8m_inference(
|
|
30 |
image=numpy.array(image),
|
31 |
object_prediction_list=prediction_result.object_prediction_list,
|
32 |
rect_th=2,
|
33 |
-
text_size=
|
34 |
)
|
35 |
|
36 |
output = Image.fromarray(visual_result["image"])
|
|
|
30 |
image=numpy.array(image),
|
31 |
object_prediction_list=prediction_result.object_prediction_list,
|
32 |
rect_th=2,
|
33 |
+
text_size=1
|
34 |
)
|
35 |
|
36 |
output = Image.fromarray(visual_result["image"])
|