README.md 7.6 KB
Newer Older
K
Kentaro Wada 已提交
1
<img src="labelme/icons/icon.png" align="right" />
K
Kentaro Wada 已提交
2

K
Kentaro Wada 已提交
3
# labelme: Image Polygonal Annotation with Python
K
Kentaro Wada 已提交
4

K
Kentaro Wada 已提交
5
[![PyPI Version](https://img.shields.io/pypi/v/labelme.svg)](https://pypi.python.org/pypi/labelme)
K
Kentaro Wada 已提交
6
[![Python Versions](https://img.shields.io/pypi/pyversions/labelme.svg)](https://pypi.org/project/labelme)
K
Kentaro Wada 已提交
7
[![Travis Build Status](https://travis-ci.org/wkentaro/labelme.svg?branch=master)](https://travis-ci.org/wkentaro/labelme)
K
Kentaro Wada 已提交
8
[![Docker Build Status](https://img.shields.io/docker/build/wkentaro/labelme.svg)](https://hub.docker.com/r/wkentaro/labelme)
K
Kentaro Wada 已提交
9 10


K
Kentaro Wada 已提交
11
Labelme is a graphical image annotation tool inspired by <http://labelme.csail.mit.edu>.  
12
It is written in Python and uses Qt for its graphical interface.
K
Kentaro Wada 已提交
13

K
Kentaro Wada 已提交
14
<img src="examples/instance_segmentation/.readme/annotation.jpg" width="80%" />
K
Kentaro Wada 已提交
15
<i>Fig 1. Annotation example of instance segmentation.</i>
K
Kentaro Wada 已提交
16

K
Kentaro Wada 已提交
17
<img src="examples/instance_segmentation/data_dataset_voc/JPEGImages/2011_000006.jpg" width="19%" /> <img src="examples/instance_segmentation/data_dataset_voc/SegmentationClassPNG/2011_000006.png" width="19%" /> <img src="examples/instance_segmentation/data_dataset_voc/SegmentationClassVisualization/2011_000006.jpg" width="19%" /> <img src="examples/instance_segmentation/data_dataset_voc/SegmentationObjectPNG/2011_000006.png" width="19%" /> <img src="examples/instance_segmentation/data_dataset_voc/SegmentationObjectVisualization/2011_000006.jpg" width="19%" />  
K
Kentaro Wada 已提交
18
<i>Fig 2. VOC dataset example of instance segmentation.</i>
K
Kentaro Wada 已提交
19

K
Kentaro Wada 已提交
20
<img src="examples/semantic_segmentation/.readme/annotation.jpg" width="30%" /> <img src="examples/bbox_detection/.readme/annotation.jpg" width="32%" /> <img src="examples/classification/.readme/annotation_cat.jpg" width="33%" />  
K
Kentaro Wada 已提交
21
<i>Fig 3. Other examples (semantic segmentation, bbox detection, and classification).</i>
K
Kentaro Wada 已提交
22

K
Kentaro Wada 已提交
23 24 25
<img src="https://user-images.githubusercontent.com/4310419/47907116-85667800-de82-11e8-83d0-b9f4eb33268f.gif" width="30%" /> <img src="https://user-images.githubusercontent.com/4310419/47922172-57972880-deae-11e8-84f8-e4324a7c856a.gif" width="30%" /> <img src="https://user-images.githubusercontent.com/14256482/46932075-92145f00-d080-11e8-8d09-2162070ae57c.png" width="32%" />  
<i>Fig 4. Various primitives (polygon, rectangle, circle, line, and point).</i>

K
Kentaro Wada 已提交
26

27 28
## Features

K
Kentaro Wada 已提交
29
- [x] Image annotation for polygon, rectangle, circle, line and point. ([tutorial](examples/tutorial))
K
Kentaro Wada 已提交
30
- [x] Image flag annotation for classification and cleaning. ([#166](https://github.com/wkentaro/labelme/pull/166))
K
Kentaro Wada 已提交
31
- [x] Video annotation. ([video annotation](examples/video_annotation))
K
Kentaro Wada 已提交
32
- [x] GUI customization (predefined labels / flags, auto-saving, label validation, etc). ([#144](https://github.com/wkentaro/labelme/pull/144))
K
Kentaro Wada 已提交
33 34
- [x] Exporting VOC-format dataset for semantic/instance segmentation. ([semantic segmentation](examples/semantic_segmentation), [instance segmentation](examples/instance_segmentation))
- [x] Exporting COCO-format dataset for instance segmentation. ([instance segmentation](examples/instance_segmentation))
K
Kentaro Wada 已提交
35

36 37


K
Kentaro Wada 已提交
38
## Requirements
K
Kentaro Wada 已提交
39

K
Kentaro Wada 已提交
40 41
- Ubuntu / macOS / Windows
- Python2 / Python3
K
Kentaro Wada 已提交
42
- [PyQt4 / PyQt5](http://www.riverbankcomputing.co.uk/software/pyqt/intro) / [PySide2](https://wiki.qt.io/PySide2_GettingStarted)
K
Kentaro Wada 已提交
43 44


K
Kentaro Wada 已提交
45
## Installation
K
Kentaro Wada 已提交
46

K
Kentaro Wada 已提交
47 48
There are options:

49 50
- Platform agonistic installation: [Anaconda](#anaconda), [Docker](#docker)
- Platform specific installation: [Ubuntu](#ubuntu), [macOS](#macos), [Windows](#windows)
K
Kentaro Wada 已提交
51

K
Kentaro Wada 已提交
52
### Anaconda
K
Kentaro Wada 已提交
53 54 55 56

You need install [Anaconda](https://www.continuum.io/downloads), then run below:

```bash
K
Kentaro Wada 已提交
57
# python2
K
Kentaro Wada 已提交
58 59
conda create --name=labelme python=2.7
source activate labelme
K
Kentaro Wada 已提交
60
# conda install -c conda-forge pyside2
K
Kentaro Wada 已提交
61 62
conda install pyqt
pip install labelme
K
Kentaro Wada 已提交
63 64
# if you'd like to use the latest version. run below:
# pip install git+https://github.com/wkentaro/labelme.git
K
Kentaro Wada 已提交
65 66 67 68

# python3
conda create --name=labelme python=3.6
source activate labelme
K
Kentaro Wada 已提交
69
# conda install -c conda-forge pyside2
K
Kentaro Wada 已提交
70
# conda install pyqt
K
Kentaro Wada 已提交
71
pip install pyqt5  # pyqt5 can be installed via pip on python3
K
Kentaro Wada 已提交
72
pip install labelme
K
Kentaro Wada 已提交
73 74
```

K
Kentaro Wada 已提交
75
### Docker
K
Kentaro Wada 已提交
76

K
Kentaro Wada 已提交
77
You need install [docker](https://www.docker.com), then run below:
K
Kentaro Wada 已提交
78 79

```bash
K
Kentaro Wada 已提交
80
wget https://raw.githubusercontent.com/wkentaro/labelme/master/labelme/cli/on_docker.py -O labelme_on_docker
K
Kentaro Wada 已提交
81 82 83
chmod u+x labelme_on_docker

# Maybe you need http://sourabhbajaj.com/blog/2017/02/07/gui-applications-docker-mac/ on macOS
K
Kentaro Wada 已提交
84 85
./labelme_on_docker examples/tutorial/apc2016_obj3.jpg -O examples/tutorial/apc2016_obj3.json
./labelme_on_docker examples/semantic_segmentation/data_annotated
K
Kentaro Wada 已提交
86
```
K
Kentaro Wada 已提交
87

K
Kentaro Wada 已提交
88
### Ubuntu
K
Kentaro Wada 已提交
89 90

```bash
K
Kentaro Wada 已提交
91
# Ubuntu 14.04 / Ubuntu 16.04
K
Kentaro Wada 已提交
92
# Python2
93 94
# sudo apt-get install python-qt4  # PyQt4
sudo apt-get install python-pyqt5  # PyQt5
K
Kentaro Wada 已提交
95
sudo pip install labelme
K
Kentaro Wada 已提交
96
# Python3
97
sudo apt-get install python3-pyqt5  # PyQt5
K
Kentaro Wada 已提交
98
sudo pip3 install labelme
K
Kentaro Wada 已提交
99 100
```

K
Kentaro Wada 已提交
101
### macOS
K
Kentaro Wada 已提交
102 103

```bash
K
Kentaro Wada 已提交
104
# macOS Sierra
K
Kentaro Wada 已提交
105 106
brew install pyqt  # maybe pyqt5
pip install labelme  # both python2/3 should work
K
Kentaro Wada 已提交
107

K
Kentaro Wada 已提交
108 109
# or install standalone executable / app
brew install wkentaro/labelme/labelme
K
Kentaro Wada 已提交
110
brew cask install wkentaro/labelme/labelme
K
Kentaro Wada 已提交
111 112
```

113 114 115 116 117 118 119 120 121 122
### Windows

Firstly, follow instruction in [Anaconda](#anaconda).

```bash
# Pillow 5 causes dll load error on Windows.
# https://github.com/wkentaro/labelme/pull/174
conda install pillow=4.0.0
```

K
Kentaro Wada 已提交
123

K
Kentaro Wada 已提交
124
## Usage
K
Kentaro Wada 已提交
125

K
Kentaro Wada 已提交
126 127
Run `labelme --help` for detail.  
The annotations are saved as a [JSON](http://www.json.org/) file.
K
Kentaro Wada 已提交
128 129

```bash
K
Kentaro Wada 已提交
130 131
labelme  # just open gui

K
Kentaro Wada 已提交
132 133
# tutorial (single image example)
cd examples/tutorial
K
Kentaro Wada 已提交
134 135 136 137 138 139 140 141 142 143
labelme apc2016_obj3.jpg  # specify image file
labelme apc2016_obj3.jpg -O apc2016_obj3.json  # close window after the save
labelme apc2016_obj3.jpg --nodata  # not include image data but relative image path in JSON file
labelme apc2016_obj3.jpg \
  --labels highland_6539_self_stick_notes,mead_index_cards,kong_air_dog_squeakair_tennis_ball  # specify label list

# semantic segmentation example
cd examples/semantic_segmentation
labelme data_annotated/  # Open directory to annotate all images in it
labelme data_annotated/ --labels labels.txt  # specify label list with a file
K
Kentaro Wada 已提交
144 145
```

K
Kentaro Wada 已提交
146
For more advanced usage, please refer to the examples:
K
Kentaro Wada 已提交
147

K
Kentaro Wada 已提交
148 149 150 151
* [Tutorial (Single Image Example)](examples/tutorial)
* [Semantic Segmentation Example](examples/semantic_segmentation)
* [Instance Segmentation Example](examples/instance_segmentation)
* [Video Annotation Example](examples/video_annotation)
K
Kentaro Wada 已提交
152 153


K
Kentaro Wada 已提交
154 155
## FAQ

K
Kentaro Wada 已提交
156 157 158 159
- **How to convert JSON file to numpy array?** See [examples/tutorial](examples/tutorial#convert-to-dataset).
- **How to load label PNG file?** See [examples/tutorial](examples/tutorial#how-to-load-label-png-file).
- **How to get annotations for semantic segmentation?** See [examples/semantic_segmentation](examples/semantic_segmentation).
- **How to get annotations for instance segmentation?** See [examples/instance_segmentation](examples/instance_segmentation).
K
Kentaro Wada 已提交
160 161


K
Kentaro Wada 已提交
162
## Screencast
K
Kentaro Wada 已提交
163

K
Kentaro Wada 已提交
164
<img src="https://drive.google.com/uc?id=1Ad-pn3mVM17eMDiMPYToVfzKFCaRQMdB" width="70%"/>
K
Kentaro Wada 已提交
165 166


K
Kentaro Wada 已提交
167
## Testing
K
Kentaro Wada 已提交
168 169 170 171 172 173 174 175

```bash
pip install hacking pytest pytest-qt
flake8 .
pytest -v tests
```


K
Kentaro Wada 已提交
176 177 178 179 180 181 182 183 184 185 186 187 188
## Developing

```bash
git clone https://github.com/wkentaro/labelme.git
cd labelme

# Install anaconda3 and labelme
curl -L https://github.com/wkentaro/dotfiles/raw/master/local/bin/install_anaconda3.sh | bash -s .
source .anaconda3/bin/activate
pip install -e .
```


K
Kentaro Wada 已提交
189
## How to build standalone executable
K
Kentaro Wada 已提交
190

K
Kentaro Wada 已提交
191
Below shows how to build the standalone executable on macOS, Linux and Windows.  
K
Kentaro Wada 已提交
192
Also, there are pre-built executables in
K
Kentaro Wada 已提交
193
[the release section](https://github.com/wkentaro/labelme/releases).
K
Kentaro Wada 已提交
194 195

```bash
K
Kentaro Wada 已提交
196
# Setup conda
K
Kentaro Wada 已提交
197
conda create --name labelme python==3.6.0
K
Kentaro Wada 已提交
198
conda activate labelme
K
Kentaro Wada 已提交
199 200 201

# Build the standalone executable
pip install .
K
Kentaro Wada 已提交
202
pip install pyinstaller
K
Kentaro Wada 已提交
203
pyinstaller labelme.spec
K
Kentaro Wada 已提交
204
dist/labelme --version
K
Kentaro Wada 已提交
205 206 207
```


K
Kentaro Wada 已提交
208
## Acknowledgement
K
Kentaro Wada 已提交
209

K
Kentaro Wada 已提交
210 211
This repo is the fork of [mpitid/pylabelme](https://github.com/mpitid/pylabelme),
whose development has already stopped.