From a1ee8304b737985867349c6dece9efc95deb4e13 Mon Sep 17 00:00:00 2001 From: Hossein Zahaki Date: Sat, 17 Feb 2024 21:42:19 +0000 Subject: [PATCH 1/5] Add dark mode toggle and qdarkstyle library --- labelImg.py | 20 +++++++++++++++++++- requirements/requirements-linux-python3.txt | 1 + 2 files changed, 20 insertions(+), 1 deletion(-) diff --git a/labelImg.py b/labelImg.py index efd8a2976..a2432d7b4 100755 --- a/labelImg.py +++ b/labelImg.py @@ -23,7 +23,8 @@ sip.setapi('QVariant', 2) from PyQt4.QtGui import * from PyQt4.QtCore import * - + +import qdarkstyle from libs.combobox import ComboBox from libs.default_label_combobox import DefaultLabelComboBox from libs.resources import * @@ -426,6 +427,14 @@ def get_format_meta(format): self.display_label_option.setChecked(settings.get(SETTING_PAINT_LABEL, False)) self.display_label_option.triggered.connect(self.toggle_paint_labels_option) + # Add a toggle action for dark/light mode + dark_mode_init = True # Set the initial mode + self.dark_mode_action = QAction('Dark Mode', self) + self.dark_mode_action.setCheckable(True) + self.dark_mode_action.setChecked(dark_mode_init) + self.dark_mode_action.triggered.connect(self.toggle_dark_mode) + self.toggle_dark_mode() if dark_mode_init else None + add_actions(self.menus.file, (open, open_dir, change_save_dir, open_annotation, copy_prev_bounding, self.menus.recentFiles, save, save_format, save_as, close, reset_all, delete_image, quit)) add_actions(self.menus.help, (help_default, show_info, show_shortcut)) @@ -433,6 +442,7 @@ def get_format_meta(format): self.auto_saving, self.single_class_mode, self.display_label_option, + self.dark_mode_action, labels, advanced_mode, None, hide_all, show_all, None, zoom_in, zoom_out, zoom_org, None, @@ -1669,6 +1679,14 @@ def toggle_paint_labels_option(self): def toggle_draw_square(self): self.canvas.set_drawing_shape_to_square(self.draw_squares_option.isChecked()) + def toggle_dark_mode(self): + if self.dark_mode_action.isChecked(): + # Apply dark mode stylesheet + self.setStyleSheet(qdarkstyle.load_stylesheet_pyqt5()) + else: + # Apply light mode stylesheet or your default stylesheet + self.setStyleSheet("") # Replace with your default stylesheet if any + def inverted(color): return QColor(*[255 - v for v in color.getRgb()]) diff --git a/requirements/requirements-linux-python3.txt b/requirements/requirements-linux-python3.txt index d27cf7b82..606f42386 100644 --- a/requirements/requirements-linux-python3.txt +++ b/requirements/requirements-linux-python3.txt @@ -1,2 +1,3 @@ pyqt5==5.14.1 lxml==4.9.1 +qdarkstyle==3.2.3 From 73f89de92b9cf2af843528726c03dfd9b68a74fe Mon Sep 17 00:00:00 2001 From: Hossein Zahaki Date: Fri, 7 Jun 2024 19:43:04 +0000 Subject: [PATCH 2/5] feat: start from last opened image --- labelImg.py | 19 +++++++++++++------ libs/constants.py | 1 + libs/settings.py | 3 ++- 3 files changed, 16 insertions(+), 7 deletions(-) diff --git a/labelImg.py b/labelImg.py index a2432d7b4..7e54ff8a0 100755 --- a/labelImg.py +++ b/labelImg.py @@ -98,7 +98,6 @@ def __init__(self, default_filename=None, default_prefdef_class_file=None, defau self.dir_name = None self.label_hist = [] self.last_open_dir = None - self.cur_img_idx = 0 self.img_count = len(self.m_img_list) # Whether we need to save or not. @@ -126,6 +125,7 @@ def __init__(self, default_filename=None, default_prefdef_class_file=None, defau list_layout = QVBoxLayout() list_layout.setContentsMargins(0, 0, 0, 0) + # Create a widget for using default label self.use_default_label_checkbox = QCheckBox(get_str('useDefaultLabel')) self.use_default_label_checkbox.setChecked(False) @@ -242,6 +242,9 @@ def __init__(self, default_filename=None, default_prefdef_class_file=None, defau save = action(get_str('save'), self.save_file, 'Ctrl+S', 'save', get_str('saveDetail'), enabled=False) + + self.cur_img_idx = settings.get(SETTING_LAST_IMAGE_INDEX, 0) + print(self.cur_img_idx) def get_format_meta(format): """ @@ -1174,8 +1177,9 @@ def load_file(self, file_path=None): # Default : select last item if there is at least one item if self.label_list.count(): - self.label_list.setCurrentItem(self.label_list.item(self.label_list.count() - 1)) - self.label_list.item(self.label_list.count() - 1).setSelected(True) + item_idx = self.label_list.count() - 1 + self.label_list.setCurrentItem(self.label_list.item(item_idx)) + self.label_list.item(item_idx).setSelected(True) self.canvas.setFocus(True) return True @@ -1284,6 +1288,8 @@ def closeEvent(self, event): settings[SETTING_PAINT_LABEL] = self.display_label_option.isChecked() settings[SETTING_DRAW_SQUARE] = self.draw_squares_option.isChecked() settings[SETTING_LABEL_FILE_FORMAT] = self.label_file_format + settings[SETTING_LAST_IMAGE_INDEX] = self.cur_img_idx + # print('save_settings') settings.save() def load_recent(self, filename): @@ -1381,6 +1387,7 @@ def import_dir_images(self, dir_path): self.file_list_widget.clear() self.m_img_list = self.scan_all_images(dir_path) self.img_count = len(self.m_img_list) + self.open_next_image() for imgPath in self.m_img_list: item = QListWidgetItem(imgPath) @@ -1450,8 +1457,8 @@ def open_next_image(self, _value=False): filename = None if self.file_path is None: - filename = self.m_img_list[0] - self.cur_img_idx = 0 + filename = self.m_img_list[self.cur_img_idx] + # self.cur_img_idx = 0 else: if self.cur_img_idx + 1 < self.img_count: self.cur_img_idx += 1 @@ -1470,7 +1477,7 @@ def open_file(self, _value=False): if filename: if isinstance(filename, (tuple, list)): filename = filename[0] - self.cur_img_idx = 0 + # self.cur_img_idx = 0 self.img_count = 1 self.load_file(filename) diff --git a/libs/constants.py b/libs/constants.py index 1efda037c..2b4f362ae 100644 --- a/libs/constants.py +++ b/libs/constants.py @@ -18,3 +18,4 @@ SETTING_DRAW_SQUARE = 'draw/square' SETTING_LABEL_FILE_FORMAT= 'labelFileFormat' DEFAULT_ENCODING = 'utf-8' +SETTING_LAST_IMAGE_INDEX = 'lastImageIndex' diff --git a/libs/settings.py b/libs/settings.py index a6e8b868b..21557fbc9 100644 --- a/libs/settings.py +++ b/libs/settings.py @@ -5,7 +5,7 @@ class Settings(object): def __init__(self): # Be default, the home will be in the same folder as labelImg - home = os.path.expanduser("~") + home = os.path.expanduser("./") self.data = {} self.path = os.path.join(home, '.labelImgSettings.pkl') @@ -31,6 +31,7 @@ def load(self): try: if os.path.exists(self.path): with open(self.path, 'rb') as f: + # print(f) self.data = pickle.load(f) return True except: From 7951cdf07e18eeffeec5ec2e17372e1280f2ca6a Mon Sep 17 00:00:00 2001 From: Hossein Zahaki Date: Fri, 7 Jun 2024 19:51:12 +0000 Subject: [PATCH 3/5] update readme.rst --- README.rst | 313 +---------------------------------------------------- 1 file changed, 4 insertions(+), 309 deletions(-) diff --git a/README.rst b/README.rst index ef061f46f..fbb633ab0 100644 --- a/README.rst +++ b/README.rst @@ -4,314 +4,9 @@ Label Studio is a modern, multi-modal data annotation tool ======= -LabelImg, the popular image annotation tool created by Tzutalin with the help of dozens contributors, is no longer actively being developed and has become part of the Label Studio community. Check out `Label Studio `__, the most flexible open source data labeling tool for images, text, hypertext, audio, video and time-series data. `Install `__ Label Studio and join the `slack community `__ to get started. -.. image:: /readme/images/label-studio-1-6-player-screenshot.png - :target: https://github.com/heartexlabs/label-studio - -About LabelImg -======== - -.. image:: https://img.shields.io/pypi/v/labelimg.svg - :target: https://pypi.python.org/pypi/labelimg - -.. image:: https://img.shields.io/github/workflow/status/tzutalin/labelImg/Package?style=for-the-badge - :alt: GitHub Workflow Status - -.. image:: https://img.shields.io/badge/lang-en-blue.svg - :target: https://github.com/tzutalin/labelImg - -.. image:: https://img.shields.io/badge/lang-zh-green.svg - :target: https://github.com/tzutalin/labelImg/blob/master/readme/README.zh.rst - -.. image:: https://img.shields.io/badge/lang-jp-green.svg - :target: https://github.com/tzutalin/labelImg/blob/master/readme/README.jp.rst - -LabelImg is a graphical image annotation tool. - -It is written in Python and uses Qt for its graphical interface. - -Annotations are saved as XML files in PASCAL VOC format, the format used -by `ImageNet `__. Besides, it also supports YOLO and CreateML formats. - -.. image:: https://raw.githubusercontent.com/tzutalin/labelImg/master/demo/demo3.jpg - :alt: Demo Image - -.. image:: https://raw.githubusercontent.com/tzutalin/labelImg/master/demo/demo.jpg - :alt: Demo Image - -`Watch a demo video `__ - -Installation ------------------- - -Get from PyPI but only python3.0 or above -~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ -This is the simplest (one-command) install method on modern Linux distributions such as Ubuntu and Fedora. - -.. code:: shell - - pip3 install labelImg - labelImg - labelImg [IMAGE_PATH] [PRE-DEFINED CLASS FILE] - - -Build from source -~~~~~~~~~~~~~~~~~ - -Linux/Ubuntu/Mac requires at least `Python -2.6 `__ and has been tested with `PyQt -4.8 `__. However, `Python -3 or above `__ and `PyQt5 `__ are strongly recommended. - - -Ubuntu Linux -^^^^^^^^^^^^ - -Python 3 + Qt5 - -.. code:: shell - - sudo apt-get install pyqt5-dev-tools - sudo pip3 install -r requirements/requirements-linux-python3.txt - make qt5py3 - python3 labelImg.py - python3 labelImg.py [IMAGE_PATH] [PRE-DEFINED CLASS FILE] - -macOS -^^^^^ - -Python 3 + Qt5 - -.. code:: shell - - brew install qt # Install qt-5.x.x by Homebrew - brew install libxml2 - - or using pip - - pip3 install pyqt5 lxml # Install qt and lxml by pip - - make qt5py3 - python3 labelImg.py - python3 labelImg.py [IMAGE_PATH] [PRE-DEFINED CLASS FILE] - - -Python 3 Virtualenv (Recommended) - -Virtualenv can avoid a lot of the QT / Python version issues - -.. code:: shell - - brew install python3 - pip3 install pipenv - pipenv run pip install pyqt5==5.15.2 lxml - pipenv run make qt5py3 - pipenv run python3 labelImg.py - [Optional] rm -rf build dist; pipenv run python setup.py py2app -A;mv "dist/labelImg.app" /Applications - -Note: The Last command gives you a nice .app file with a new SVG Icon in your /Applications folder. You can consider using the script: build-tools/build-for-macos.sh - - -Windows -^^^^^^^ - -Install `Python `__, -`PyQt5 `__ -and `install lxml `__. - -Open cmd and go to the `labelImg <#labelimg>`__ directory - -.. code:: shell - - pyrcc4 -o libs/resources.py resources.qrc - For pyqt5, pyrcc5 -o libs/resources.py resources.qrc - - python labelImg.py - python labelImg.py [IMAGE_PATH] [PRE-DEFINED CLASS FILE] - -If you want to package it into a separate EXE file - -.. code:: shell - - Install pyinstaller and execute: - - pip install pyinstaller - pyinstaller --hidden-import=pyqt5 --hidden-import=lxml -F -n "labelImg" -c labelImg.py -p ./libs -p ./ - -Windows + Anaconda -^^^^^^^^^^^^^^^^^^ - -Download and install `Anaconda `__ (Python 3+) - -Open the Anaconda Prompt and go to the `labelImg <#labelimg>`__ directory - -.. code:: shell - - conda install pyqt=5 - conda install -c anaconda lxml - pyrcc5 -o libs/resources.py resources.qrc - python labelImg.py - python labelImg.py [IMAGE_PATH] [PRE-DEFINED CLASS FILE] - -Use Docker -~~~~~~~~~~~~~~~~~ -.. code:: shell - - docker run -it \ - --user $(id -u) \ - -e DISPLAY=unix$DISPLAY \ - --workdir=$(pwd) \ - --volume="/home/$USER:/home/$USER" \ - --volume="/etc/group:/etc/group:ro" \ - --volume="/etc/passwd:/etc/passwd:ro" \ - --volume="/etc/shadow:/etc/shadow:ro" \ - --volume="/etc/sudoers.d:/etc/sudoers.d:ro" \ - -v /tmp/.X11-unix:/tmp/.X11-unix \ - tzutalin/py2qt4 - - make qt4py2;./labelImg.py - -You can pull the image which has all of the installed and required dependencies. `Watch a demo video `__ - - -Usage ------ - -Steps (PascalVOC) -~~~~~~~~~~~~~~~~~ - -1. Build and launch using the instructions above. -2. Click 'Change default saved annotation folder' in Menu/File -3. Click 'Open Dir' -4. Click 'Create RectBox' -5. Click and release left mouse to select a region to annotate the rect - box -6. You can use right mouse to drag the rect box to copy or move it - -The annotation will be saved to the folder you specify. - -You can refer to the below hotkeys to speed up your workflow. - -Steps (YOLO) -~~~~~~~~~~~~ - -1. In ``data/predefined_classes.txt`` define the list of classes that will be used for your training. - -2. Build and launch using the instructions above. - -3. Right below "Save" button in the toolbar, click "PascalVOC" button to switch to YOLO format. - -4. You may use Open/OpenDIR to process single or multiple images. When finished with a single image, click save. - -A txt file of YOLO format will be saved in the same folder as your image with same name. A file named "classes.txt" is saved to that folder too. "classes.txt" defines the list of class names that your YOLO label refers to. - -Note: - -- Your label list shall not change in the middle of processing a list of images. When you save an image, classes.txt will also get updated, while previous annotations will not be updated. - -- You shouldn't use "default class" function when saving to YOLO format, it will not be referred. - -- When saving as YOLO format, "difficult" flag is discarded. - -Create pre-defined classes -~~~~~~~~~~~~~~~~~~~~~~~~~~ - -You can edit the -`data/predefined\_classes.txt `__ -to load pre-defined classes - -Annotation visualization -~~~~~~~~~~~~~~~~~~~~~~~~ - -1. Copy the existing lables file to same folder with the images. The labels file name must be same with image file name. - -2. Click File and choose 'Open Dir' then Open the image folder. - -3. Select image in File List, it will appear the bounding box and label for all objects in that image. - -(Choose Display Labels mode in View to show/hide lablels) - - -Hotkeys -~~~~~~~ - -+--------------------+--------------------------------------------+ -| Ctrl + u | Load all of the images from a directory | -+--------------------+--------------------------------------------+ -| Ctrl + r | Change the default annotation target dir | -+--------------------+--------------------------------------------+ -| Ctrl + s | Save | -+--------------------+--------------------------------------------+ -| Ctrl + d | Copy the current label and rect box | -+--------------------+--------------------------------------------+ -| Ctrl + Shift + d | Delete the current image | -+--------------------+--------------------------------------------+ -| Space | Flag the current image as verified | -+--------------------+--------------------------------------------+ -| w | Create a rect box | -+--------------------+--------------------------------------------+ -| d | Next image | -+--------------------+--------------------------------------------+ -| a | Previous image | -+--------------------+--------------------------------------------+ -| del | Delete the selected rect box | -+--------------------+--------------------------------------------+ -| Ctrl++ | Zoom in | -+--------------------+--------------------------------------------+ -| Ctrl-- | Zoom out | -+--------------------+--------------------------------------------+ -| ↑→↓← | Keyboard arrows to move selected rect box | -+--------------------+--------------------------------------------+ - -**Verify Image:** - -When pressing space, the user can flag the image as verified, a green background will appear. -This is used when creating a dataset automatically, the user can then through all the pictures and flag them instead of annotate them. - -**Difficult:** - -The difficult field is set to 1 indicates that the object has been annotated as "difficult", for example, an object which is clearly visible but difficult to recognize without substantial use of context. -According to your deep neural network implementation, you can include or exclude difficult objects during training. - -How to reset the settings -~~~~~~~~~~~~~~~~~~~~~~~~~ - -In case there are issues with loading the classes, you can either: - -1. From the top menu of the labelimg click on Menu/File/Reset All -2. Remove the `.labelImgSettings.pkl` from your home directory. In Linux and Mac you can do: - `rm ~/.labelImgSettings.pkl` - - -How to contribute -~~~~~~~~~~~~~~~~~ - -Send a pull request - -License -~~~~~~~ -`Free software: MIT license `_ - -Citation: Tzutalin. LabelImg. Git code (2015). https://github.com/tzutalin/labelImg - -Related and additional tools -~~~~~~~~~~~~~~~~~~~~~~~~~~~~ - -1. `Label Studio `__ to label images, text, audio, video and time-series data for machine learning and AI -2. `ImageNet Utils `__ to - download image, create a label text for machine learning, etc -3. `Use Docker to run labelImg `__ -4. `Generating the PASCAL VOC TFRecord files `__ -5. `App Icon based on Icon by Nick Roach (GPL) `__ -6. `Setup python development in vscode `__ -7. `The link of this project on iHub platform `__ -8. `Convert annotation files to CSV format or format for Google Cloud AutoML `__ - - - -Stargazers over time -~~~~~~~~~~~~~~~~~~~~ - -.. image:: https://starchart.cc/tzutalin/labelImg.svg +The `labelImg.py` code now includes: +- **Dark Mode**: Toggleable from the View menu for low-light use. +- **Last Image Memory**: Remembers the last image index for easier restarts. +These enhancements enhance user experience and workflow efficiency. \ No newline at end of file From 01ab459bf599edadc01966f72094dcde8f94f14a Mon Sep 17 00:00:00 2001 From: MrZahaki Date: Fri, 7 Jun 2024 19:52:45 +0000 Subject: [PATCH 4/5] Update README.rst --- README.rst | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/README.rst b/README.rst index fbb633ab0..fb1b05776 100644 --- a/README.rst +++ b/README.rst @@ -6,7 +6,9 @@ Label Studio is a modern, multi-modal data annotation tool The `labelImg.py` code now includes: + - **Dark Mode**: Toggleable from the View menu for low-light use. + - **Last Image Memory**: Remembers the last image index for easier restarts. -These enhancements enhance user experience and workflow efficiency. \ No newline at end of file +These enhancements enhance user experience and workflow efficiency. From 0bcab5a80b263d385bd4432d2fbf1276f323da82 Mon Sep 17 00:00:00 2001 From: MrZahaki Date: Fri, 7 Jun 2024 20:24:30 +0000 Subject: [PATCH 5/5] Update README.rst --- README.rst | 7 ------- 1 file changed, 7 deletions(-) diff --git a/README.rst b/README.rst index fb1b05776..13bceb2bb 100644 --- a/README.rst +++ b/README.rst @@ -1,10 +1,3 @@ -.. image:: /readme/images/labelimg.png - :target: https://github.com/heartexlabs/label-studio - -Label Studio is a modern, multi-modal data annotation tool -======= - - The `labelImg.py` code now includes: - **Dark Mode**: Toggleable from the View menu for low-light use.