INSTALL.md 3.4 KB
Newer Older
Y
Yang Zhang 已提交
1
# Installation
K
Kaipeng Deng 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14

---
## Table of Contents

- [Introduction](#introduction)
- [PaddlePaddle](#paddlepaddle)
- [Other Dependencies](#other-dependencies)
- [PaddleDetection](#paddle-detection)
- [Datasets](#datasets)


## Introduction

Y
Yang Zhang 已提交
15 16
This document covers how to install PaddleDetection, its dependencies
(including PaddlePaddle), together with COCO and PASCAL VOC dataset.
K
Kaipeng Deng 已提交
17 18 19 20 21 22

For general information about PaddleDetection, please see [README.md](../README.md).


## PaddlePaddle

Y
Yang Zhang 已提交
23
Running PaddleDetection requires PaddlePaddle Fluid v.1.5 and later. please follow the instructions in [installation document](http://www.paddlepaddle.org/documentation/docs/en/1.4/beginners_guide/install/index_en.html).
K
Kaipeng Deng 已提交
24

Y
Yang Zhang 已提交
25 26
Please make sure your PaddlePaddle installation was successful and the version
of your PaddlePaddle is not lower than required. Verify with the following commands.
K
Kaipeng Deng 已提交
27 28 29 30 31

```
# To check if PaddlePaddle installation was sucessful
python -c "from paddle.fluid import fluid; fluid.install_check.run_check()"

Y
Yang Zhang 已提交
32
# To check PaddlePaddle version
K
Kaipeng Deng 已提交
33 34 35 36 37 38 39
python -c "import paddle; print(paddle.__version__)"
```

### Requirements:

- Python2 or Python3
- CUDA >= 8.0
40
- cuDNN >= 5.0
K
Kaipeng Deng 已提交
41 42 43 44 45
- nccl >= 2.1.2


## Other Dependencies

Y
Yang Zhang 已提交
46
[COCO-API](https://github.com/cocodataset/cocoapi):
K
Kaipeng Deng 已提交
47

Y
Yang Zhang 已提交
48
COCO-API is needed for training. Installation is as follows:
K
Kaipeng Deng 已提交
49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64

    git clone https://github.com/cocodataset/cocoapi.git
    cd cocoapi/PythonAPI
    # if cython is not installed
    pip install Cython
    # Install into global site-packages
    make install
    # Alternatively, if you do not have permissions or prefer
    # not to install the COCO API into global site-packages
    python setup.py install --user


## PaddleDetection

**Clone Paddle models repository:**

Y
Yang Zhang 已提交
65 66
You can clone Paddle models and change working directory to PaddleDetection
with the following commands:
K
Kaipeng Deng 已提交
67 68 69 70 71 72 73

```
cd <path/to/clone/models>
git clone https://github.com/PaddlePaddle/models
cd models/PaddleCV/object_detection
```

Y
Yang Zhang 已提交
74
**Install Python dependencies:**
K
Kaipeng Deng 已提交
75

Y
Yang Zhang 已提交
76
Required python packages are specified in [requirements.txt](./requirements.txt), and can be installed with:
K
Kaipeng Deng 已提交
77 78 79 80 81

```
pip install -r requirements.txt
```

Y
Yang Zhang 已提交
82
**Make sure the tests pass:**
K
Kaipeng Deng 已提交
83 84

```
85
export PYTHONPATH=`pwd`:$PYTHONPATH
K
Kaipeng Deng 已提交
86 87 88 89 90 91
python ppdet/modeling/tests/test_architectures.py
```


## Datasets

Y
Yang Zhang 已提交
92
PaddleDetection includes support for [MSCOCO](http://cocodataset.org) and [PASCAL VOC](http://host.robots.ox.ac.uk/pascal/VOC/) by default, please follow these instructions to set up the dataset.
K
Kaipeng Deng 已提交
93

Y
Yang Zhang 已提交
94
**Create symlinks for local datasets:**
K
Kaipeng Deng 已提交
95

Y
Yang Zhang 已提交
96 97 98
Default dataset path in config files is `data/coco` and `data/voc`, if the
datasets are already available on disk, you can simply create symlinks to
their directories:
K
Kaipeng Deng 已提交
99 100

```
Y
Yang Zhang 已提交
101 102
ln -sf <path/to/coco> <path/to/paddle_detection>/data/coco
ln -sf <path/to/voc> <path/to/paddle_detection>/data/voc
K
Kaipeng Deng 已提交
103 104
```

Y
Yang Zhang 已提交
105 106 107
**Download datasets manually:**

On the other hand, to download the datasets, run the following commands:
K
Kaipeng Deng 已提交
108

109
- MS-COCO
K
Kaipeng Deng 已提交
110 111

```
112 113
cd dataset/coco
./download.sh
K
Kaipeng Deng 已提交
114 115
```

116
- PASCAL VOC
K
Kaipeng Deng 已提交
117 118

```
119 120
cd dataset/voc
./download.sh
K
Kaipeng Deng 已提交
121 122
```

Y
Yang Zhang 已提交
123
**Download datasets automatically:**
K
Kaipeng Deng 已提交
124

Y
Yang Zhang 已提交
125 126 127 128 129 130
If a training session is started but the dataset is not setup properly (e.g,
not found in `data/coc` or `data/voc`), PaddleDetection can automatically
download them from [MSCOCO-2017](http://images.cocodataset.org) and
[VOC2012](http://host.robots.ox.ac.uk/pascal/VOC), the decompressed datasets
will be cached in `~/.cache/paddle/dataset/` and can be discovered automatically
subsequently.
K
Kaipeng Deng 已提交
131 132


Y
Yang Zhang 已提交
133
**NOTE:** For further informations on the datasets, please see [DATA.md](DATA.md)