From 27fe3b23d6a387b31976b6bf7de1bba6490b99ce Mon Sep 17 00:00:00 2001 From: Stortebeker <52454983+Stortebeker@users.noreply.github.com> Date: Sat, 18 Sep 2021 13:17:35 +0300 Subject: [PATCH 1/2] Update README.md --- README.md | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index ee521fe..4356c08 100644 --- a/README.md +++ b/README.md @@ -52,9 +52,13 @@ Example usage: python main.py train \ --style ./path/to/style/image.jpg \ --dataset ./path/to/dataset \ - --weights ./path/to/weights \ + --weights ./path/to/weights (--weights ~/.keras/models/vgg19_weights_tf_dim_ordering_tf_kernels_notop.h5) \ --batch 2 +##### Note: The vgg19 weights are downloaded on first run in ~/.keras/models/ . + +##### Note: The training output is saved in project directory . + ### Evaluating Style Transfer Networks Use `main.py` to evaluate a style transfer network. Evaluation takes 2s per frame(712x474) on a GTX 1060 3GB. **Models for evaluation are [located here.](https://drive.google.com/drive/folders/1-ywa__KcK4uEEYOzgfeRCpCzP3RJKBwL?usp=sharing)** @@ -62,7 +66,7 @@ Evaluation takes 2s per frame(712x474) on a GTX 1060 3GB. **Models for evaluati Example usage: python main.py evaluate \ - --weights ./path/to/weights \ + --weights ./path/to/weights (use the .index file) \ --content ./path/to/content/image.jpg(video.mp4) \ --result ./path/to/save/results/image.jpg From be6e3fbc980f91f42a8d639718ed5e9c25493bfc Mon Sep 17 00:00:00 2001 From: Stortebeker <52454983+Stortebeker@users.noreply.github.com> Date: Sat, 18 Sep 2021 14:44:28 +0300 Subject: [PATCH 2/2] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 4356c08..d6e4543 100644 --- a/README.md +++ b/README.md @@ -52,7 +52,7 @@ Example usage: python main.py train \ --style ./path/to/style/image.jpg \ --dataset ./path/to/dataset \ - --weights ./path/to/weights (--weights ~/.keras/models/vgg19_weights_tf_dim_ordering_tf_kernels_notop.h5) \ + --weights ./path/to/weights \ --batch 2 ##### Note: The vgg19 weights are downloaded on first run in ~/.keras/models/ .