Spaces:
Running
on
Zero
Running
on
Zero
Update object_description_generator.py
Browse files
object_description_generator.py
CHANGED
@@ -20,7 +20,7 @@ class ObjectDescriptionGenerator:
|
|
20 |
min_prominence_score: float = 0.1,
|
21 |
max_categories_to_return: int = 5,
|
22 |
max_total_objects: int = 7,
|
23 |
-
confidence_threshold_for_description: float = 0.
|
24 |
region_analyzer: Optional[Any] = None):
|
25 |
"""
|
26 |
初始化物件描述生成器
|
@@ -604,7 +604,7 @@ class ObjectDescriptionGenerator:
|
|
604 |
|
605 |
for processed_pos in processed_positions:
|
606 |
position_distance = abs(obj_position[0] - processed_pos[0]) + abs(obj_position[1] - processed_pos[1])
|
607 |
-
if position_distance < 0.
|
608 |
is_duplicate = True
|
609 |
break
|
610 |
|
|
|
20 |
min_prominence_score: float = 0.1,
|
21 |
max_categories_to_return: int = 5,
|
22 |
max_total_objects: int = 7,
|
23 |
+
confidence_threshold_for_description: float = 0.25,
|
24 |
region_analyzer: Optional[Any] = None):
|
25 |
"""
|
26 |
初始化物件描述生成器
|
|
|
604 |
|
605 |
for processed_pos in processed_positions:
|
606 |
position_distance = abs(obj_position[0] - processed_pos[0]) + abs(obj_position[1] - processed_pos[1])
|
607 |
+
if position_distance < 0.15:
|
608 |
is_duplicate = True
|
609 |
break
|
610 |
|