Update object_description_generator.py

object_description_generator.py

@@ -592,30 +592,30 @@ class ObjectDescriptionGenerator:
                             return (priority, -len(obj_group_list), -avg_area, -quantity_bonus)
 
                         # remove duplicate
-                        # deduplicated_objects_by_class = {}
-                        # processed_positions = []
+                        deduplicated_objects_by_class = {}
+                        processed_positions = []
 
-                        # for class_name, group_of_objects in objects_by_class.items():
-                        #     unique_objects = []
+                        for class_name, group_of_objects in objects_by_class.items():
+                            unique_objects = []
 
-                        #     for obj in group_of_objects:
-                        #         obj_position = obj.get("normalized_center", [0.5, 0.5])
-                        #         is_duplicate = False
+                            for obj in group_of_objects:
+                                obj_position = obj.get("normalized_center", [0.5, 0.5])
+                                is_duplicate = False
 
-                        #         for processed_pos in processed_positions:
-                        #             position_distance = abs(obj_position[0] - processed_pos[0]) + abs(obj_position[1] - processed_pos[1])
-                        #             if position_distance < 0.15:
-                        #                 is_duplicate = True
-                        #                 break
+                                for processed_pos in processed_positions:
+                                    position_distance = abs(obj_position[0] - processed_pos[0]) + abs(obj_position[1] - processed_pos[1])
+                                    if position_distance < 0.15:
+                                        is_duplicate = True
+                                        break
 
-                        #         if not is_duplicate:
-                        #             unique_objects.append(obj)
-                        #             processed_positions.append(obj_position)
+                                if not is_duplicate:
+                                    unique_objects.append(obj)
+                                    processed_positions.append(obj_position)
 
-                        #     if unique_objects:
-                        #         deduplicated_objects_by_class[class_name] = unique_objects
+                            if unique_objects:
+                                deduplicated_objects_by_class[class_name] = unique_objects
 
-                        # objects_by_class = deduplicated_objects_by_class
+                        objects_by_class = deduplicated_objects_by_class
                         
                         print(f"DEBUG: After spatial deduplication:")
                         for class_name in ["car", "traffic light", "person", "handbag"]: