DawnC commited on
Commit
e3868ba
·
verified ·
1 Parent(s): 5003ee4

Update object_extractor.py

Browse files
Files changed (1) hide show
  1. object_extractor.py +4 -4
object_extractor.py CHANGED
@@ -25,15 +25,15 @@ class ObjectExtractor:
25
  self.object_categories = object_categories or {}
26
 
27
  # 1. 讀取並設定基本信心度門檻(如果外部沒傳,就預設 0.25)
28
- self.base_conf_threshold = 0.2
29
 
30
  # 2. 動態信心度調整映射表 (key: 小寫 class_name, value: 調整係數)
31
  # 最終的門檻 = base_conf_threshold * factor
32
  # 如果某個 class_name 沒在這裡,就直接用 base_conf_threshold(相當於 factor=1.0)
33
  self.dynamic_conf_map = {
34
- "traffic light": 0.8,
35
- "car": 0.9,
36
- "person": 0.8,
37
 
38
  }
39
 
 
25
  self.object_categories = object_categories or {}
26
 
27
  # 1. 讀取並設定基本信心度門檻(如果外部沒傳,就預設 0.25)
28
+ self.base_conf_threshold = 0.25
29
 
30
  # 2. 動態信心度調整映射表 (key: 小寫 class_name, value: 調整係數)
31
  # 最終的門檻 = base_conf_threshold * factor
32
  # 如果某個 class_name 沒在這裡,就直接用 base_conf_threshold(相當於 factor=1.0)
33
  self.dynamic_conf_map = {
34
+ "traffic light": 0.6,
35
+ "car": 0.8,
36
+ "person": 0.7,
37
 
38
  }
39