README.md 9.9 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
  <a href="https://hub.docker.com/r/wkentaro/labelme"><img src="https://img.shields.io/docker/cloud/build/wkentaro/labelme"></a>
K
Kentaro Wada 已提交
14
</div>
K
Kentaro Wada 已提交
15

K
Kentaro Wada 已提交
16 17 18 19 20 21 22 23
<div align="center">
  <a href="#installation"><b>Installation</b></a> |
  <a href="#usage"><b>Usage</b></a> |
  <a href="https://github.com/wkentaro/labelme/tree/master/examples/tutorial#tutorial-single-image-example"><b>Tutorial</b></a> |
  <a href="https://github.com/wkentaro/labelme/tree/master/examples"><b>Examples</b></a> |
  <a href="https://www.youtube.com/playlist?list=PLI6LvFw0iflh3o33YYnVIfOpaO0hc5Dzw"><b>Youtube FAQ</b></a>
</div>

K
Kentaro Wada 已提交
24 25 26
<br/>

<div align="center">
K
Kentaro Wada 已提交
27
  <img src="examples/instance_segmentation/.readme/annotation.jpg" width="70%">
K
Kentaro Wada 已提交
28 29 30
</div>

## Description
K
Kentaro Wada 已提交
31

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

K
Kentaro Wada 已提交
35
<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 已提交
36
<i>VOC dataset example of instance segmentation.</i>
K
Kentaro Wada 已提交
37

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

K
Kentaro Wada 已提交
41
<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 已提交
42
<i>Various primitives (polygon, rectangle, circle, line, and point).</i>
K
Kentaro Wada 已提交
43

K
Kentaro Wada 已提交
44

45 46
## Features

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

54 55


K
Kentaro Wada 已提交
56
## Requirements
K
Kentaro Wada 已提交
57

K
Kentaro Wada 已提交
58 59
- Ubuntu / macOS / Windows
- Python2 / Python3
K
Kentaro Wada 已提交
60
- [PyQt4 / PyQt5](http://www.riverbankcomputing.co.uk/software/pyqt/intro)
K
Kentaro Wada 已提交
61 62


K
Kentaro Wada 已提交
63
## Installation
K
Kentaro Wada 已提交
64

K
Kentaro Wada 已提交
65 66
There are options:

I
ilayluz 已提交
67
- Platform agnostic installation: [Anaconda](#anaconda), [Docker](#docker)
68
- Platform specific installation: [Ubuntu](#ubuntu), [macOS](#macos), [Windows](#windows)
K
Kentaro Wada 已提交
69
- Pre-build binaries from [the release section](https://github.com/wkentaro/labelme/releases)
K
Kentaro Wada 已提交
70

K
Kentaro Wada 已提交
71
### Anaconda
K
Kentaro Wada 已提交
72 73 74 75

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

```bash
K
Kentaro Wada 已提交
76
# python2
K
Kentaro Wada 已提交
77 78
conda create --name=labelme python=2.7
source activate labelme
K
Kentaro Wada 已提交
79
# conda install -c conda-forge pyside2
K
Kentaro Wada 已提交
80 81
conda install pyqt
pip install labelme
K
Kentaro Wada 已提交
82 83
# if you'd like to use the latest version. run below:
# pip install git+https://github.com/wkentaro/labelme.git
K
Kentaro Wada 已提交
84 85 86 87

# python3
conda create --name=labelme python=3.6
source activate labelme
K
Kentaro Wada 已提交
88
# conda install -c conda-forge pyside2
K
Kentaro Wada 已提交
89
# conda install pyqt
K
Kentaro Wada 已提交
90
# pip install pyqt5  # pyqt5 can be installed via pip on python3
K
Kentaro Wada 已提交
91
pip install labelme
K
Kentaro Wada 已提交
92 93
# or you can install everything by conda command
# conda install labelme -c conda-forge
K
Kentaro Wada 已提交
94 95
```

K
Kentaro Wada 已提交
96
### Docker
K
Kentaro Wada 已提交
97

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

```bash
K
Kentaro Wada 已提交
101 102 103
# on macOS
socat TCP-LISTEN:6000,reuseaddr,fork UNIX-CLIENT:\"$DISPLAY\" &
docker run -it -v /tmp/.X11-unix:/tmp/.X11-unix -e DISPLAY=docker.for.mac.host.internal:0 -v $(pwd):/root/workdir wkentaro/labelme
K
Kentaro Wada 已提交
104

K
Kentaro Wada 已提交
105 106 107
# on Linux
xhost +
docker run -it -v /tmp/.X11-unix:/tmp/.X11-unix -e DISPLAY=:0 -v $(pwd):/root/workdir wkentaro/labelme
K
Kentaro Wada 已提交
108
```
K
Kentaro Wada 已提交
109

K
Kentaro Wada 已提交
110
### Ubuntu
K
Kentaro Wada 已提交
111 112

```bash
K
Kentaro Wada 已提交
113
# Ubuntu 14.04 / Ubuntu 16.04
K
Kentaro Wada 已提交
114
# Python2
115 116
# sudo apt-get install python-qt4  # PyQt4
sudo apt-get install python-pyqt5  # PyQt5
K
Kentaro Wada 已提交
117
sudo pip install labelme
K
Kentaro Wada 已提交
118
# Python3
119
sudo apt-get install python3-pyqt5  # PyQt5
K
Kentaro Wada 已提交
120
sudo pip3 install labelme
K
Kentaro Wada 已提交
121 122 123

# or install standalone executable from:
# https://github.com/wkentaro/labelme/releases
K
Kentaro Wada 已提交
124 125
```

A
Alex Myczko 已提交
126 127 128 129 130 131
### Ubuntu 19.10+ / Debian (sid)

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

K
Kentaro Wada 已提交
132
### macOS
K
Kentaro Wada 已提交
133 134

```bash
K
Kentaro Wada 已提交
135
# macOS Sierra
K
Kentaro Wada 已提交
136 137
brew install pyqt  # maybe pyqt5
pip install labelme  # both python2/3 should work
K
Kentaro Wada 已提交
138

K
Kentaro Wada 已提交
139 140
# or install standalone executable/app from:
# https://github.com/wkentaro/labelme/releases
K
Kentaro Wada 已提交
141 142
```

143 144
### Windows

145
Install [Anaconda](https://www.continuum.io/downloads), then in an Anaconda Prompt run:
146 147

```bash
148 149 150 151
# python3
conda create --name=labelme python=3.6
conda activate labelme
pip install labelme
152 153
```

K
Kentaro Wada 已提交
154

K
Kentaro Wada 已提交
155
## Usage
K
Kentaro Wada 已提交
156

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

```bash
K
Kentaro Wada 已提交
161 162
labelme  # just open gui

K
Kentaro Wada 已提交
163 164
# tutorial (single image example)
cd examples/tutorial
K
Kentaro Wada 已提交
165 166 167 168 169 170 171 172 173 174
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 已提交
175 176
```

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

K
Kentaro Wada 已提交
179 180 181 182
* [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 已提交
183

C
Christian Clauss 已提交
184
### Command Line Arguments
R
Roger Iyengar 已提交
185 186 187 188 189
- `--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 已提交
190

K
Kentaro Wada 已提交
191 192
## FAQ

K
Kentaro Wada 已提交
193 194 195 196
- **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 已提交
197 198


K
Kentaro Wada 已提交
199
## Testing
K
Kentaro Wada 已提交
200 201 202 203 204 205 206 207

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


K
Kentaro Wada 已提交
208 209 210 211 212 213 214 215 216 217 218 219 220
## 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 已提交
221
## How to build standalone executable
K
Kentaro Wada 已提交
222

K
Kentaro Wada 已提交
223
Below shows how to build the standalone executable on macOS, Linux and Windows.  
K
Kentaro Wada 已提交
224 225

```bash
K
Kentaro Wada 已提交
226
# Setup conda
K
Kentaro Wada 已提交
227
conda create --name labelme python==3.6.0
K
Kentaro Wada 已提交
228
conda activate labelme
K
Kentaro Wada 已提交
229 230 231

# Build the standalone executable
pip install .
K
Kentaro Wada 已提交
232
pip install pyinstaller
K
Kentaro Wada 已提交
233
pyinstaller labelme.spec
K
Kentaro Wada 已提交
234
dist/labelme --version
K
Kentaro Wada 已提交
235 236 237
```


238 239 240 241 242 243 244 245 246 247 248 249 250 251
## How to contribute

Make sure below test passes on your environment.  
See `.github/workflows/ci.yml` for more detail.

```bash
pip install black hacking pytest pytest-qt

flake8 .
black --line-length 79 --check labelme/
MPLBACKEND='agg' pytest tests/ -m 'not gpu'
```


K
Kentaro Wada 已提交
252
## Acknowledgement
K
Kentaro Wada 已提交
253

K
Kentaro Wada 已提交
254 255
This repo is the fork of [mpitid/pylabelme](https://github.com/mpitid/pylabelme),
whose development has already stopped.
K
Kentaro Wada 已提交
256 257 258 259 260 261 262 263


## 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,
K
Kentaro Wada 已提交
264
  author =       {Kentaro Wada},
K
Kentaro Wada 已提交
265 266 267 268 269
  title =        {{labelme: Image Polygonal Annotation with Python}},
  howpublished = {\url{https://github.com/wkentaro/labelme}},
  year =         {2016}
}
```