new

Author: Pimfayette-lab

PyImageLabeling/config.json

@@ -2,6 +2,7 @@
     "window_size":{
         "margin": 10,
         "icon": [20, 20],
+        "icon_tool": [30, 30],
         "icon_save_marker": [12, 12],
 
         "graphics_view":{
@@ -155,6 +156,7 @@
                     {
                         "name": "contour_filling",
                         "icon": "filling",
+                        "icon_tool": "filling_tool",
                         "checkable": true,
                         "name_view": "Contour Filling",
                         "tooltip": "Click to activate Move mode. This allows you to move the image around in the viewer by dragging it."
@@ -164,13 +166,15 @@
                         "name_view": "Paintbrush",
                         "checkable": true,
                         "icon": "paint",
+                        "icon_tool": "paint_tool",
                         "tooltip": "Click to reset the image's position and zoom level to the default view."
                     },
                     {
                         "name": "magic_pen",
                         "name_view": "Magic Pen",
                         "checkable": true,
                         "icon": "magic",
+                        "icon_tool": "magi_tool",
                         "tooltip": "Click to activate Move mode. This allows you to move the image around in the viewer by dragging it."
                     }
                 ]
@@ -217,6 +221,7 @@
                     "name_view": "Eraser",
                     "checkable": true,
                     "icon": "eraser",
+                    "icon_tool": "eraser_tool",
                     "tooltip": "Click to activate the eraser tool, allowing you to erase parts of the image or layer."
                 },
                 {

PyImageLabeling/icons/eraser_tool.png

@@ -239,7 +239,19 @@ def save(self, current_file_path, path_image):
         name, format = name.split(".")
         format = "png"
         save_file = current_file_path + os.sep+name + KEYWORD_SAVE_LABEL + str(self.label.get_label_id()) + "." + format 
-        self.labeling_overlay_pixmap.save(save_file, format)
+        image = self.labeling_overlay_pixmap.toImage().convertToFormat(QImage.Format.Format_Grayscale8)
+
+        # Apply binary threshold
+        threshold = 1  # any nonzero pixel will become white
+        for y in range(image.height()):
+            for x in range(image.width()):
+                pixel_val = image.pixelColor(x, y).value()  # grayscale intensity
+                if pixel_val > threshold:
+                    image.setPixel(x, y, 0xFFFFFFFF)  # white
+                else:
+                    image.setPixel(x, y, 0xFF000000)  # black
+
+        image.save(save_file, format.upper())
 
     def remove_save(self, current_file_path, path_image):
         name = os.path.basename(path_image)

PyImageLabeling/icons/filling_tool.png

@@ -35,8 +35,13 @@ def save(self):
             dialog.setOption(QFileDialog.Option.ShowDirsOnly, False)  
             dialog.setOption(QFileDialog.Option.DontUseNativeDialog, True)  
             dialog.setViewMode(QFileDialog.ViewMode.Detail)
-            if dialog.exec():
-                default_path = dialog.selectedFiles()[0]
+            dialog.setDirectory(default_path)
+            dialog.setModal(True)
+            if dialog.exec() == 0: return 
+            #print("result:", result)
+            #if dialog.exec():
+            #print("default_path:", default_path)
+            default_path = dialog.selectedFiles()[0]
             current_file_path = default_path
 
             if len(current_file_path) == 0: return

PyImageLabeling/icons/magic_tool.png

@@ -1 +1 @@
-{"contour_filling": {"tolerance": 8}, "magic_pen": {"tolerance": 72, "max_pixels": 408780, "method": "RGB"}, "save": {"path": "/home/nicolas/Bureau/ProjetsMAIA/Picels/test"}, "load": {"path": "/home/nicolas/Bureau/ProjetsMAIA/Picels/raw_data", "alpha_color": [255, 255, 255]}, "zoom": {"min_zoom": 0.5, "max_zoom": 100, "plus_zoom_factor": 1.1, "minus_zoom_factor": 0.9}, "paint_brush": {"size": 34}, "eraser": {"size": 12, "absolute_mode": 0}, "labeling_opacity": 45}
+{"contour_filling": {"tolerance": 8}, "magic_pen": {"tolerance": 72, "max_pixels": 408780, "method": "RGB"}, "save": {"path": "/home/nicolas/Bureau/ProjetsMAIA/Picels/test"}, "load": {"path": "C:/Users/pimfa/Pictures", "alpha_color": [255, 255, 255]}, "zoom": {"min_zoom": 0.5, "max_zoom": 100, "plus_zoom_factor": 1.1, "minus_zoom_factor": 0.9}, "paint_brush": {"size": 34}, "eraser": {"size": 12, "absolute_mode": 0}, "labeling_opacity": 45}

PyImageLabeling/icons/paint_tool.png

@@ -60,8 +60,19 @@ def __init__(self, view, parent=None):
         self.view.minus_zoom_factor = self.data_parameters["zoom"]["minus_zoom_factor"]
 
         self.view.layer_activation = False
-        
+    
     def change_cursor(self, name):
+            try:
+                cursor_pixmap = QPixmap(Utils.get_icon_path(f"{name}_tool"))
+                cursor_pixmap = cursor_pixmap.scaled(*self.view.config["window_size"]["icon_tool"])
+            except:
+                cursor_pixmap = QPixmap(Utils.get_icon_path(name))
+                cursor_pixmap = cursor_pixmap.scaled(*self.view.config["window_size"]["icon"])
+            cursor = QCursor(cursor_pixmap)
+            self.viewport().setCursor(cursor)
+            return cursor.pixmap().width(), cursor.pixmap().height()
+
+    def change_cursor_n(self, name):
         cursor_pixmap = QPixmap(Utils.get_icon_path(name))
         cursor_pixmap = cursor_pixmap.scaled(*self.view.config["window_size"]["icon"]) 
         cursor = QCursor(cursor_pixmap)