Merge branch 'master' of https://github.com/sunshineatnoon/MaskStyleT…

…ransfer

README.md

@@ -8,6 +8,9 @@
 ### Optional
 - [kaggle-cli](https://github.com/floydwch/kaggle-cli) : Used to download WikiArt by command line
 
+## Pre-trained VGG & Decoder
+Download from [Google drive](https://drive.google.com/file/d/1REga1z1rKezQtBebIZ86_iNR-mxum-KB/view?usp=sharing).
+
 ## DATASET
 ### MSCOCO
 ```
@@ -23,7 +26,7 @@ kg download -u <username> -p <password> -c painter-by-numbers -f train.zip
 ## Training
 
 ```
-python Train.py --vgg_dir PATH_TO_VGG --decoder_dir PATH_TO_DECODER --layer r31 --contentPath PATH_TO_MSCOCO --stylePath PATH_TO_WikiArt --reg_weight 100 --outf OUTPUT_DIR
+python Train.py --vgg_dir PATH_TO_VGG/vgg_normalised_conv3_1.t7 --decoder_dir PATH_TO_DECODER/feature_invertor_conv3_1.t7 --layer r31 --contentPath PATH_TO_MSCOCO --stylePath PATH_TO_WikiArt --reg_weight 100 --outf OUTPUT_DIR
 ```
 Intermediate results and weight will be stored in `OUTPUT_DIR`