Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
set cpu default
  • Loading branch information
kuaikuaikim authored Dec 28, 2017
1 parent 319ef58 commit 2130014
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ Our inspiration comes from several research papers on this topic, as well as cur
## Installation

DFace has two major module, detection and recognition.In these two, We provide all tutorials about how to train a model and running.
First setting a pytorch and cv2. We suggest Anaconda to make a virtual and independent python envirment.
First setting a pytorch and cv2. We suggest Anaconda to make a virtual and independent python envirment.**If you want to train on GPU,please install Nvidia cuda and cudnn.**

### Requirements
* cuda 8.0
Expand Down Expand Up @@ -144,7 +144,9 @@ python dface/prepare_data/assemble_onet_imglist.py
python dface/train_net/train_o_net.py
```

#### Test face detection
#### Test face detection
**If you don't want to train,you can request me to get model parameters file, then put onet_epoch.pt,pnet_epoch.pt,rnet_epoch.pt in model_store folder.**

```shell
python test_image.py
```
Expand All @@ -154,7 +156,7 @@ python test_image.py
TODO


## Demo
## Demo  

![mtcnn](http://pic.dface.io/figure_2.png)

Expand Down

0 comments on commit 2130014

Please sign in to comment.