Spaces:
Running
on
Zero
Running
on
Zero
Update object_extractor.py
Browse files- 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.
|
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.
|
35 |
-
"car": 0.
|
36 |
-
"person": 0.
|
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 |
|