rein0421 commited on
Commit
ebd489e
·
verified ·
1 Parent(s): c979fb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -17
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
- 'License plate': 0.3,
87
- 'Mail or envelope': 0.3,
 
 
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
- tex = [
210
- 'text','Name tag', 'License plate', 'Mail', 'Documents', 'QR codes',
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) / 10) # 個数決定(1/2)
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
  # マスクを生成する関数