README.md 9.3 KB
Newer Older
K
Kentaro Wada 已提交
1 2 3
<h1 align="center">
  <img src="labelme/icons/icon.png"><br/>labelme
</h1>
K
Kentaro Wada 已提交
4

K
Kentaro Wada 已提交
5 6 7
<h4 align="center">
  Image Polygonal Annotation with Python
</h4>
K
Kentaro Wada 已提交
8

K
Kentaro Wada 已提交
9 10 11
<div align="center">
  <a href="https://pypi.python.org/pypi/labelme"><img src="https://img.shields.io/pypi/v/labelme.svg"></a>
  <a href="https://pypi.org/project/labelme"><img src="https://img.shields.io/pypi/pyversions/labelme.svg"></a>
K
Kentaro Wada 已提交
12
  <a href="https://github.com/wkentaro/labelme/actions"><img src="https://github.com/wkentaro/labelme/workflows/CI/badge.svg?branch=master&event=push"></a>
K
Kentaro Wada 已提交
13 14
  <a href="https://hub.docker.com/r/wkentaro/labelme"><img src="https://img.shields.io/docker/build/wkentaro/labelme.svg"></a>
</div>
K
Kentaro Wada 已提交
15

K
Kentaro Wada 已提交
16 17 18
<br/>

<div align="center">
K
Kentaro Wada 已提交
19
  <img src="examples/instance_segmentation/.readme/annotation.jpg" width="70%">
K
Kentaro Wada 已提交
20 21 22
</div>

## Description
K
Kentaro Wada 已提交
23

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

K
Kentaro Wada 已提交
27
<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 已提交
28
<i>VOC dataset example of instance segmentation.</i>
K
Kentaro Wada 已提交
29

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

K
Kentaro Wada 已提交
33
<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%" />  
K
Kentaro Wada 已提交
34
<i>Various primitives (polygon, rectangle, circle, line, and point).</i>
K
Kentaro Wada 已提交
35

K
Kentaro Wada 已提交
36

37 38
## Features

K
Kentaro Wada 已提交
39
- [x] Image annotation for polygon, rectangle, circle, line and point. ([tutorial](examples/tutorial))
K
Kentaro Wada 已提交
40
- [x] Image flag annotation for classification and cleaning. ([#166](https://github.com/wkentaro/labelme/pull/166))
K
Kentaro Wada 已提交
41
- [x] Video annotation. ([video annotation](examples/video_annotation))
K
Kentaro Wada 已提交
42
- [x] GUI customization (predefined labels / flags, auto-saving, label validation, etc). ([#144](https://github.com/wkentaro/labelme/pull/144))
K
Kentaro Wada 已提交
43 44
- [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 已提交
45

46 47


K
Kentaro Wada 已提交
48
## Requirements
K
Kentaro Wada 已提交
49

K
Kentaro Wada 已提交
50 51
- Ubuntu / macOS / Windows
- Python2 / Python3
K
Kentaro Wada 已提交
52
- [PyQt4 / PyQt5](http://www.riverbankcomputing.co.uk/software/pyqt/intro) / [PySide2](https://wiki.qt.io/PySide2_GettingStarted)
K
Kentaro Wada 已提交
53 54


K
Kentaro Wada 已提交
55
## Installation
K
Kentaro Wada 已提交
56

K
Kentaro Wada 已提交
57 58
There are options:

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

K
Kentaro Wada 已提交
62
### Anaconda
K
Kentaro Wada 已提交
63 64 65 66

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

```bash
K
Kentaro Wada 已提交
67
# python2
K
Kentaro Wada 已提交
68 69
conda create --name=labelme python=2.7
source activate labelme
K
Kentaro Wada 已提交
70
# conda install -c conda-forge pyside2
K
Kentaro Wada 已提交
71 72
conda install pyqt
pip install labelme
K
Kentaro Wada 已提交
73 74
# if you'd like to use the latest version. run below:
# pip install git+https://github.com/wkentaro/labelme.git
K
Kentaro Wada 已提交
75 76 77 78

# python3
conda create --name=labelme python=3.6
source activate labelme
K
Kentaro Wada 已提交
79
# conda install -c conda-forge pyside2
K
Kentaro Wada 已提交
80
# conda install pyqt
K
Kentaro Wada 已提交
81
# pip install pyqt5  # pyqt5 can be installed via pip on python3
K
Kentaro Wada 已提交
82
pip install labelme
K
Kentaro Wada 已提交
83 84
# or you can install everything by conda command
# conda install labelme -c conda-forge
K
Kentaro Wada 已提交
85 86
```

K
Kentaro Wada 已提交
87
### Docker
K
Kentaro Wada 已提交
88

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

```bash
K
Kentaro Wada 已提交
92
wget https://raw.githubusercontent.com/wkentaro/labelme/master/labelme/cli/on_docker.py -O labelme_on_docker
K
Kentaro Wada 已提交
93 94 95
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 已提交
96 97
./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 已提交
98
```
K
Kentaro Wada 已提交
99

K
Kentaro Wada 已提交
100
### Ubuntu
K
Kentaro Wada 已提交
101 102

```bash
K
Kentaro Wada 已提交
103
# Ubuntu 14.04 / Ubuntu 16.04
K
Kentaro Wada 已提交
104
# Python2
105 106
# sudo apt-get install python-qt4  # PyQt4
sudo apt-get install python-pyqt5  # PyQt5
K
Kentaro Wada 已提交
107
sudo pip install labelme
K
Kentaro Wada 已提交
108
# Python3
109
sudo apt-get install python3-pyqt5  # PyQt5
K
Kentaro Wada 已提交
110
sudo pip3 install labelme
K
Kentaro Wada 已提交
111 112
```

A
Alex Myczko 已提交
113 114 115 116 117 118
### Ubuntu 19.10+ / Debian (sid)

```bash
sudo apt-get install labelme
```

K
Kentaro Wada 已提交
119
### macOS
K
Kentaro Wada 已提交
120 121

```bash
K
Kentaro Wada 已提交
122
# macOS Sierra
K
Kentaro Wada 已提交
123 124
brew install pyqt  # maybe pyqt5
pip install labelme  # both python2/3 should work
K
Kentaro Wada 已提交
125

K
Kentaro Wada 已提交
126
# or install standalone executable / app
127
# NOTE: this only installs the `labelme` command
K
Kentaro Wada 已提交
128
brew install wkentaro/labelme/labelme
K
Kentaro Wada 已提交
129
brew cask install wkentaro/labelme/labelme
K
Kentaro Wada 已提交
130 131
```

132 133 134 135 136 137 138 139 140 141
### 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 已提交
142

K
Kentaro Wada 已提交
143
## Usage
K
Kentaro Wada 已提交
144

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

```bash
K
Kentaro Wada 已提交
149 150
labelme  # just open gui

K
Kentaro Wada 已提交
151 152
# tutorial (single image example)
cd examples/tutorial
K
Kentaro Wada 已提交
153 154 155 156 157 158 159 160 161 162
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 已提交
163 164
```

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

K
Kentaro Wada 已提交
167 168 169 170
* [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 已提交
171

C
Christian Clauss 已提交
172
### Command Line Arguments
R
Roger Iyengar 已提交
173 174 175 176 177
- `--output` specifies the location that annotations will be written to. If the location ends with .json, a single annotation will be written to this file. Only one image can be annotated if a location is specified with .json. If the location does not end with .json, the program will assume it is a directory. Annotations will be stored in this directory with a name that corresponds to the image that the annotation was made on.
- The first time you run labelme, it will create a config file in `~/.labelmerc`. You can edit this file and the changes will be applied the next time that you launch labelme. If you would prefer to use a config file from another location, you can specify this file with the `--config` flag.
- Without the `--nosortlabels` flag, the program will list labels in alphabetical order. When the program is run with this flag, it will display labels in the order that they are provided.
- Flags are assigned to an entire image. [Example](examples/classification)
- Labels are assigned to a single polygon. [Example](examples/bbox_detection)
K
Kentaro Wada 已提交
178

K
Kentaro Wada 已提交
179 180
## FAQ

K
Kentaro Wada 已提交
181 182 183 184
- **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 已提交
185 186


K
Kentaro Wada 已提交
187
## Testing
K
Kentaro Wada 已提交
188 189 190 191 192 193 194 195

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


K
Kentaro Wada 已提交
196 197 198 199 200 201 202 203 204 205 206 207 208
## 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 已提交
209
## How to build standalone executable
K
Kentaro Wada 已提交
210

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

```bash
K
Kentaro Wada 已提交
216
# Setup conda
K
Kentaro Wada 已提交
217
conda create --name labelme python==3.6.0
K
Kentaro Wada 已提交
218
conda activate labelme
K
Kentaro Wada 已提交
219 220 221

# Build the standalone executable
pip install .
K
Kentaro Wada 已提交
222
pip install pyinstaller
K
Kentaro Wada 已提交
223
pyinstaller labelme.spec
K
Kentaro Wada 已提交
224
dist/labelme --version
K
Kentaro Wada 已提交
225 226 227
```


K
Kentaro Wada 已提交
228
## Acknowledgement
K
Kentaro Wada 已提交
229

K
Kentaro Wada 已提交
230 231
This repo is the fork of [mpitid/pylabelme](https://github.com/mpitid/pylabelme),
whose development has already stopped.
K
Kentaro Wada 已提交
232 233 234 235 236 237 238 239 240 241 242 243 244 245


## Cite This Project

If you use this project in your research or wish to refer to the baseline results published in the README, please use the following BibTeX entry.

```bash
@misc{labelme2016,
  author =       {Ketaro Wada},
  title =        {{labelme: Image Polygonal Annotation with Python}},
  howpublished = {\url{https://github.com/wkentaro/labelme}},
  year =         {2016}
}
```