Update object_description_generator.py

object_description_generator.py

@@ -513,6 +513,11 @@ class ObjectDescriptionGenerator:
                 # 使用置信度過濾
                 confident_objects = [obj for obj in detected_objects
                                    if obj.get("confidence", 0) >= self.confidence_threshold_for_description]
+                
+                print(f"DEBUG: After confidence filtering (threshold={self.confidence_threshold_for_description}):")                
+                for class_name in ["car", "traffic light", "person", "handbag"]:                   
+                    class_objects = [obj for obj in confident_objects if obj.get("class_name") == class_name]
+                    print(f"DEBUG: {class_name}: {len(class_objects)} confident objects")
 
                 if not confident_objects:
                     no_confident_obj_msg = "While some elements might be present, no objects were identified with sufficient confidence for a detailed description."
@@ -552,6 +557,10 @@ class ObjectDescriptionGenerator:
                             if name not in objects_by_class:
                                 objects_by_class[name] = []
                             objects_by_class[name].append(obj)
+                            print(f"DEBUG: Before spatial deduplication:")
+                            for class_name in ["car", "traffic light", "person", "handbag"]:
+                                if class_name in objects_by_class:                                    
+                                    print(f"DEBUG: {class_name}: {len(objects_by_class[class_name])} objects before dedup")
 
                     if not objects_by_class:
                         description_segments.append("No common objects were confidently identified for detailed description.")
@@ -607,12 +616,22 @@ class ObjectDescriptionGenerator:
                                 deduplicated_objects_by_class[class_name] = unique_objects
 
                         objects_by_class = deduplicated_objects_by_class
+                        
+                        print(f"DEBUG: After spatial deduplication:")
+                        for class_name in ["car", "traffic light", "person", "handbag"]:
+                            if class_name in objects_by_class:                                
+                                print(f"DEBUG: {class_name}: {len(objects_by_class[class_name])} objects after dedup")
+                            
                         sorted_object_groups = sorted(objects_by_class.items(), key=sort_key_object_groups)
 
                         object_clauses = []
 
                         for class_name, group_of_objects in sorted_object_groups:
                             count = len(group_of_objects)
+                            
+                            if class_name in ["car", "traffic light", "person", "handbag"]:                                
+                                print(f"DEBUG: Final count for {class_name}: {count}")
+                                
                             if count == 0:
                                 continue