YOLO Annotator
Visual annotation tool for creating YOLO object detection datasets in VS Code.
Features
- Visual Annotation: Draw bounding boxes directly on images
- YOLO Format: Export annotations in standard YOLO
.txt format
- Class Management: Support for custom class definitions via
classes.txt
- Keyboard Shortcuts: Efficient workflow with hotkeys
- Batch Processing: Navigate through image directories seamlessly
- Real-time Editing: Modify existing annotations with live preview
Installation
- Open VS Code
- Go to Extensions (Ctrl+Shift+X)
- Search "YOLO Annotator"
- Click Install
Usage
Quick Start
- Right Click on any image and select "Open with YOLO Annotator"
- Make sure to have a
classes.txt file somewhere in the file structure shown below.
- If labels are found you should see labels drawn on your image.
- Use the navigation panel on the top-right or keyboard shortcuts to browse through images.
Keyboard Shortcuts
A - Add new label (drawing mode)
←/→ - Navigate between images
Ctrl+S - Save current labels
Delete/Backspace - Delete selected label
Esc - Cancel drawing mode
File Structure
project/
├── images/
│ ├── image1.jpg
│ ├── image2.jpg
│ └── ...
├── labels/
│ ├── image1.txt
│ ├── image2.txt
│ └── ...
└── classes.txt
person
car
bicycle
dog
cat
Each .txt file contains one line per object:
class_id center_x center_y width height
0 0.5 0.5 0.3 0.4
1 0.2 0.3 0.1 0.2
Requirements
- VS Code 1.100.0 or higher
- Image files in JPG, PNG, or JPEG format
Contributing
- Fork the repository
- Create a feature branch
- Make your changes
- Submit a pull request
License
MIT License - see LICENSE file for details.
Support
Report issues at: https://github.com/TanvirTaaha/yolo-annotator-vscode/issues
| |