update readme

This commit is contained in:
deeppomf 2018-02-11 01:00:15 -05:00
parent 167a27e7a4
commit 3c9e5841e9

View File

@ -48,7 +48,7 @@ Decensored images will be saved to the "decensor_output_images" directory. Paste
## II. Train the pretrained model on custom dataset
Put the your custom dataset for training the "data/images" directory and convert images to npy format.
Put your custom dataset for training in the "data/images" directory and convert images to npy format.
```
$ cd training_data
@ -61,8 +61,6 @@ Train pretrained model on your custom dataset.
$ python train.py
```
The dataset will not be released. I do not want to risk trouble for distributing copyrighted pornographic material.
# To do
- Add Python 3 compatibility
- Add a user interface