Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -83,14 +83,16 @@ dangerarray=[10,30,90,50,80,20,40,70,100,60]#ここに各クラスターの危
|
|
83 |
|
84 |
thresholds = {
|
85 |
'text': 0.1,
|
86 |
-
'
|
87 |
-
'
|
|
|
|
|
88 |
'Documents': 0.3,
|
89 |
-
'digital screens': 0.6,
|
90 |
'information board': 0.5,
|
91 |
-
'signboard': 0.3,
|
92 |
'poster': 0.8,
|
93 |
'sign': 0.3,
|
|
|
|
|
94 |
'logo': 0.3,
|
95 |
'cardboard': 0.6,
|
96 |
'manhole': 0.6,
|
@@ -205,21 +207,12 @@ from ultralytics import YOLO
|
|
205 |
|
206 |
# オブジェクト選択を動的に行う関数
|
207 |
def decide_to_object(risk_level):
|
208 |
-
|
209 |
-
|
210 |
-
|
211 |
-
'barcodes', 'Map', 'Digital screens', 'information board',
|
212 |
-
'signboard', 'poster', 'sign', 'utility pole'
|
213 |
]
|
214 |
-
'''
|
215 |
-
tex = [
|
216 |
-
'text', 'License plate', 'Digital screens',
|
217 |
-
'signboard', 'poster', 'sign', 'logo', 'card', 'window', 'mirror',
|
218 |
-
'Famous landmark', 'cardboard', 'manhole', 'utility pole'
|
219 |
-
]
|
220 |
-
|
221 |
# この配列の要素の順番を変えると消える順番が変わる。
|
222 |
-
risk_level = int(risk_level / 20) * (len(tex) /
|
223 |
return tex[:int(risk_level) + 1]
|
224 |
|
225 |
# マスクを生成する関数
|
|
|
83 |
|
84 |
thresholds = {
|
85 |
'text': 0.1,
|
86 |
+
'Name tag':0.2,
|
87 |
+
'License plate': 0.1,
|
88 |
+
'digital screens': 0.4,
|
89 |
+
'signboard': 0.3,
|
90 |
'Documents': 0.3,
|
|
|
91 |
'information board': 0.5,
|
|
|
92 |
'poster': 0.8,
|
93 |
'sign': 0.3,
|
94 |
+
'information board':0.3,
|
95 |
+
'Mail or envelope': 0.3,
|
96 |
'logo': 0.3,
|
97 |
'cardboard': 0.6,
|
98 |
'manhole': 0.6,
|
|
|
207 |
|
208 |
# オブジェクト選択を動的に行う関数
|
209 |
def decide_to_object(risk_level):
|
210 |
+
tex = [
|
211 |
+
'text', 'poster','Name tag','License plate', 'digital screens',
|
212 |
+
'signboard','sign', 'logo', 'manhole', 'electricity pole','cardboard'
|
|
|
|
|
213 |
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
214 |
# この配列の要素の順番を変えると消える順番が変わる。
|
215 |
+
risk_level = int(risk_level / 20) * (len(tex) / 5) # 個数決定(1/2)
|
216 |
return tex[:int(risk_level) + 1]
|
217 |
|
218 |
# マスクを生成する関数
|