An image segmentation GUI for generating ML ready mask tensors and annotations.
Project description
LazyLabel
AI-Assisted Image Segmentation for Machine Learning Dataset Preparation
LazyLabel combines Meta's Segment Anything Model (SAM) with comprehensive manual annotation tools to accelerate the creation of pixel-perfect segmentation masks for computer vision applications.
Quick Start
pip install lazylabel-gui
lazylabel-gui
From source:
git clone https://github.com/dnzckn/LazyLabel.git
cd LazyLabel
pip install -e .
lazylabel-gui
Requirements: Python 3.10+, 8GB RAM, ~2.5GB disk space (for model weights)
Core Features
Annotation Tools
- AI (SAM): Single-click segmentation with point-based refinement (SAM 1.0 & 2.1, GPU/CPU). Use negative points to subtract regions from the prediction.
- Polygon: Vertex-level drawing and editing for precise boundaries
- Box: Bounding box annotations for object detection. Hold Shift on release to erase with the box instead of adding.
- Select: Click to select existing masks for editing, reclassing, or deletion. Hold Shift+Space to erase the overlap of a drawn segment from the selected mask.
Annotation Modes
- Single View: Fine-tune individual masks with maximum precision
- Multi View: Annotate up to 2 images simultaneously, ideal for objects in similar positions with slight variations
- Sequence: Propagate a refined mask across thousands of frames using SAM 2's video predictor
Image Processing
- FFT filtering: Remove noise and enhance edges
- Channel thresholding: Isolate objects by color
- Border cropping: Zero out pixels outside defined regions in saved outputs
- View adjustments: Brightness, contrast, gamma correction, color saturation
Export Formats
Select one or more formats from Settings. All formats can be loaded back into LazyLabel.
NPZ - One-hot encoded mask tensors (.npz)
import numpy as np
data = np.load('image.npz')
mask = data['mask'] # Shape: (height, width, num_classes)
# Each channel represents one class
sky = mask[:, :, 0]
boats = mask[:, :, 1]
cats = mask[:, :, 2]
dogs = mask[:, :, 3]
Standard Formats
| Format | Output File | Description |
|---|---|---|
| YOLO Detection | image.txt |
Bounding boxes: class_id cx cy w h (normalized) |
| YOLO Segmentation | image_seg.txt |
Polygon vertices: class_id x1 y1 x2 y2 ... (normalized) |
| COCO JSON | image_coco.json |
Per-image COCO format with polygon segmentation, bounding boxes, and area |
| Pascal VOC | image.xml |
XML bounding box annotations |
| CreateML | image_createml.json |
Apple CreateML JSON with center-based bounding boxes |
COCO supercategories: Set a class alias to name.supercategory (e.g. dog.animal) to populate the supercategory field in COCO JSON output.
Model Setup
SAM 1.0 models are downloaded automatically on first use.
If the automatic download doesn't work, you can manually download and place the model:
SAM 1.0
SAM 1.0 only requires the model weights file, no additional package installation needed.
- Download
sam_vit_h_4b8939.pthfrom the SAM repository - Place in LazyLabel's models folder:
- Via pip:
<site-packages>/lazylabel/models/(runpython -c "import lazylabel; print(lazylabel.__path__[0])"to find it) - From source:
src/lazylabel/models/
- Via pip:
SAM 2.1 (improved accuracy, required for Sequence mode)
SAM 2.1 requires both the sam2 package installed and the model weights file, since it relies on config files bundled with the package.
- Install SAM 2:
pip install git+https://github.com/facebookresearch/sam2.git - Download a model (e.g.,
sam2.1_hiera_large.pt) from the SAM 2 repository - Place in LazyLabel's models folder:
- Via pip:
<site-packages>/lazylabel/models/(runpython -c "import lazylabel; print(lazylabel.__path__[0])"to find it) - From source:
src/lazylabel/models/
- Via pip:
Select the model from the dropdown in settings.
Building Windows Executable
Create a standalone Windows executable with bundled models for offline use:
Requirements:
- Windows (native, not WSL)
- Python 3.10+
- PyInstaller:
pip install pyinstaller
Build steps:
git clone https://github.com/dnzckn/LazyLabel.git
cd LazyLabel
python build_system/windows/build_windows.py
The executable will be created in dist/LazyLabel/. The entire folder (~7-8GB) can be moved anywhere and runs offline.
Documentation
- Usage Manual - Comprehensive feature guide
- Architecture Guide - Technical implementation details
- Changelog - Version history and release notes
- GitHub Issues - Report bugs or request features
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file lazylabel_gui-1.7.0.tar.gz.
File metadata
- Download URL: lazylabel_gui-1.7.0.tar.gz
- Upload date:
- Size: 249.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
3c33d1cd30b65a8519add0da0a006ec9f8e1f56af434200a2ab95e722c2ad482
|
|
| MD5 |
83b5cba018637fdcf840281b4f8c7cbb
|
|
| BLAKE2b-256 |
78a71b61cf398e51791c83fbda44674eb364cc9bb86fc58fa030fc517e4d4781
|
File details
Details for the file lazylabel_gui-1.7.0-py3-none-any.whl.
File metadata
- Download URL: lazylabel_gui-1.7.0-py3-none-any.whl
- Upload date:
- Size: 292.3 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
c3fac084d411a185a6c466baa3e97005304a38ffe328db422164b26bff93b829
|
|
| MD5 |
a5a6adcac3c43a7fa6ca4068a5ff8789
|
|
| BLAKE2b-256 |
ea86d56ad234d5058d6cfe16a56ddf068a67668942b6a63542b486be2b8f2701
|