1. added init file to both widget & utils subdirectory, 2. modified the export file to handel previous commit {29cc0c6}. 3: removed one issues

Author: amish0

labelvim/color_dialog.py renamed to labelvim/Backup/color_dialog.py

@@ -187,22 +187,31 @@ def _generate_yolov5_image_labels_and_masks(self, file_name):
                         cv2.putText(mask, self.label_list[label], (bbox[0], bbox[1]), cv2.FONT_HERSHEY_SIMPLEX, 1, color, 2)
                 elif self.annotation_type == ANNOTATION_TYPE.POLYGON:
                     segmentation = annotation['segmentation']
-                    yolov5_segmentation = self._convert_to_yolov5_segmentation_format(segmentation, image_width, image_height)
-                    label_txt.write(f"{label} {yolov5_segmentation}\n")
+                    color = random_colors_palette[label].tolist()
+                    instance_color = random.choice(random_colors_palette.tolist())
+                    for poly in segmentation:
+                        yolov5_segmentation = self._convert_to_yolov5_segmentation_format(poly, image_width, image_height)
+                        label_txt.write(f"{label} {yolov5_segmentation}\n")
+                        poly = np.array(poly).reshape(-1, 2)
+                        poly = poly.astype(np.int32)
+                        if self.include_mask:
+                           
+                            cv2.fillPoly(mask, [poly], color)
+                            # cv2.rectangle(mask, (bbox[0], bbox[1]), (bbox[0]+bbox[2], bbox[1]+bbox[3]), color, 2)
+                            # cv2.putText(mask, self.label_list[label], (bbox[0], bbox[1]), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2)
+                        
+                        if self.include_instance:
+                            cv2.fillPoly(instance_image, [poly], instance_color)
+                        # cv2.rectangle(instance_image, (bbox[0], bbox[1]), (bbox[0]+bbox[2], bbox[1]+bbox[3]), color, 2)
+                        # cv2.putText(instance_image, self.label_list[label], (bbox[0], bbox[1]), cv2.FONT_HERSHEY_SIMPLEX, 1, (0,0,0), 2)
                     bbox = annotation['bbox']
-                    segmentation = np.array(segmentation).reshape(-1, 2)
-                    segmentation = segmentation.astype(np.int32)
+                    # bbox = annotation['bbox']
                     if self.include_mask:
-                        color = random_colors_palette[label].tolist()
-                        cv2.fillPoly(mask, [segmentation], color)
-                        # cv2.rectangle(mask, (bbox[0], bbox[1]), (bbox[0]+bbox[2], bbox[1]+bbox[3]), color, 2)
-                        # cv2.putText(mask, self.label_list[label], (bbox[0], bbox[1]), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2)
-                    
+                        cv2.rectangle(mask, (bbox[0], bbox[1]), (bbox[0]+bbox[2], bbox[1]+bbox[3]), color, 2)
                     if self.include_instance:
-                        color = random.choice(random_colors_palette.tolist())
-                        cv2.fillPoly(instance_image, [segmentation], color)
-                        # cv2.rectangle(instance_image, (bbox[0], bbox[1]), (bbox[0]+bbox[2], bbox[1]+bbox[3]), color, 2)
-                        # cv2.putText(instance_image, self.label_list[label], (bbox[0], bbox[1]), cv2.FONT_HERSHEY_SIMPLEX, 1, (0,0,0), 2)
+                        # bbox = annotation['bbox']
+                        cv2.rectangle(instance_image, (bbox[0], bbox[1]), (bbox[0]+bbox[2], bbox[1]+bbox[3]), instance_color, 2)
+                
             if self.include_img:
                 # cv2.fillPoly(image, [segmentation], (0, 0, 255))
                 # cv2.rectangle(image, (bbox[0], bbox[1]), (bbox[0]+bbox[2], bbox[1]+bbox[3]), (0, 0, 255), 2)
@@ -453,14 +462,22 @@ def _generate_masks_and_images(self, annotation_data, image_info):
             instance_image = np.zeros((image_info["height"], image_info["width"],3), dtype=np.uint8)
 
         for annotation in annotation_data['annotations']:
-            segmentation = np.array(annotation['segmentation']).reshape(-1, 2)
-            segmentation = segmentation.astype(np.int32)
+            color = random_colors_palette[annotation["category_id"]].tolist()
+            instance_color = random.choice(random_colors_palette.tolist())
+            for poly in annotation['segmentation']:
+                segmentation = np.array(poly).reshape(-1, 2)
+                segmentation = segmentation.astype(np.int32)
+                if self.include_mask:
+                    cv2.fillPoly(mask, [segmentation], color)
+                if self.include_instance:
+                    cv2.fillPoly(instance_image, [segmentation], instance_color)
+            bbox = annotation['bbox']
             if self.include_mask:
-                color = random_colors_palette[annotation["category_id"]].tolist()
-                cv2.fillPoly(mask, [segmentation], 255)
+                cv2.rectangle(mask, (bbox[0], bbox[1]), (bbox[0]+bbox[2], bbox[1]+bbox[3]), color, 2)
             if self.include_instance:
-                color = random.choice(random_colors_palette.tolist())
-                cv2.fillPoly(instance_image, [segmentation], color)
+                # bbox = annotation['bbox']
+                cv2.rectangle(instance_image, (bbox[0], bbox[1]), (bbox[0]+bbox[2], bbox[1]+bbox[3]), instance_color, 2)
+                
 
         if self.include_img:
             image = cv2.addWeighted(image, 0.5, mask, 0.5, 0)

labelvim/Layout.ui renamed to labelvim/Designer/Layout.ui

@@ -14,6 +14,7 @@ def __init__(self, items: list, data: list, annotation_type: Enum, update_label_
         self.annotation_type = annotation_type
         self.selected_item = None
         self.selected_index = None
+        self.text_filter = False
         self.select_id = -1
         self.id = [-1]
         self.layout = QVBoxLayout(self)
@@ -28,6 +29,7 @@ def __init__(self, items: list, data: list, annotation_type: Enum, update_label_
             line_edit_id_combo_layout.addWidget(label)
             self.id_combo = QComboBox(self)
             self.update_id_combo()
+            self.id_combo.currentIndexChanged.connect(self.update_list_widget)
             # self.id_combo.addItems([str(i) for i in self.id])
             line_edit_id_combo_layout.addWidget(self.id_combo)
         self.layout.addLayout(line_edit_id_combo_layout)
@@ -58,6 +60,7 @@ def update_id_combo(self):
         self.id_combo.addItems(self.id)
 
     def filter_items(self, text):
+        self.text_filter = True
         """Filter the items in the list based on the text."""
         self.list_widget.clear()
         filtered_items = [item for item in self.items if text.lower() in item.lower()]
@@ -67,6 +70,25 @@ def filter_items(self, text):
             self.id = [str(-1)]
             self.id = self.id + [str(i['id']) for item in filtered_items for i in self.data if item.lower() == self.items[i["category_id"]].lower()]
             self.id_combo.addItems(self.id)
+        self.text_filter = False
+    
+    def update_list_widget(self):
+        if not self.text_filter:
+            self.list_widget.clear()
+            if self.id_combo.currentText() == str(-1):
+                self.list_widget.addItems(self.items)
+            else:
+                id = int(self.id_combo.currentText())
+                print(f" id in labepuop: {id}")
+                for i in self.data:
+                    if i["id"] == id:
+                        category_id = int(i["category_id"])
+                        break
+
+            # category_id = int(self.data[id]["category_id"])
+
+            self.list_widget.addItems([self.items[category_id]])
+            # self.list_widget.addItems([item for item in self.items if item.lower() == self.items[int(self.id_combo.currentText())].lower()])
 
 
     def add_item(self):