falseu
commited on
Commit
·
341b820
1
Parent(s):
e4050d7
update readme
Browse files
README.md
CHANGED
|
@@ -43,10 +43,10 @@ optional arguments:
|
|
| 43 |
|
| 44 |
Download [vgg19 weight](https://drive.google.com/file/d/1UcSl-Zn3byEmn15NIPXMf9zaGCKc2gfx/view?usp=sharing), [decoder weight](https://drive.google.com/file/d/18JpLtMOapA-vwBz-LRomyTl24A9GwhTF/view?usp=sharing) under main folder.
|
| 45 |
|
| 46 |
-
To test basic style transfer, run the script test_image.py. Specify `--content_image` or
|
| 47 |
|
| 48 |
```
|
| 49 |
-
$ python test.py --content_image $IMG --style_image $STYLE --
|
| 50 |
|
| 51 |
optional arguments:
|
| 52 |
-h, --help show this help message and exit
|
|
@@ -58,7 +58,7 @@ optional arguments:
|
|
| 58 |
single style image
|
| 59 |
--style_dir STYLE_DIR
|
| 60 |
style image directory, iterate all images under this directory
|
| 61 |
-
--decoder_weight DECODER_WEIGHT decoder weight file
|
| 62 |
--alpha {Alpha Range}
|
| 63 |
Alpha [0.0, 1.0] controls style transfer level
|
| 64 |
--cuda Use CUDA
|
|
@@ -69,10 +69,10 @@ optional arguments:
|
|
| 69 |
|
| 70 |
### Test Image Interpolation Style Transfer
|
| 71 |
|
| 72 |
-
To test style transfer interpolation, run the script test_interpolate.py. Specify
|
| 73 |
|
| 74 |
```
|
| 75 |
-
$ python test_interpolation.py --content_image $IMG --style_image $STYLE
|
| 76 |
|
| 77 |
optional arguments:
|
| 78 |
-h, --help show this help message and exit
|
|
@@ -80,9 +80,9 @@ optional arguments:
|
|
| 80 |
single content image file
|
| 81 |
--style_image STYLE_IMAGE
|
| 82 |
multiple style images file separated by comma
|
| 83 |
-
--decoder_weight DECODER_WEIGHT decoder weight file
|
| 84 |
--alpha {Alpha Range}
|
| 85 |
-
Alpha [0.0, 1.0] controls style transfer level
|
| 86 |
--interpolation_weights INTERPOLATION_WEIGHTS
|
| 87 |
Interpolation weight of each style image, separated by comma.
|
| 88 |
Do not specify if input grid_pth.
|
|
@@ -99,7 +99,7 @@ optional arguments:
|
|
| 99 |
To test video style transfer, run the script test_video.py.
|
| 100 |
|
| 101 |
```
|
| 102 |
-
$ python test_video.py --content_video $VID --style_image $STYLE --
|
| 103 |
|
| 104 |
optional arguments:
|
| 105 |
-h, --help show this help message and exit
|
|
@@ -107,12 +107,23 @@ optional arguments:
|
|
| 107 |
single content video file
|
| 108 |
--style_image STYLE_IMAGE
|
| 109 |
single style image
|
| 110 |
-
--decoder_weight DECODER_WEIGHT decoder weight file
|
| 111 |
--alpha {Alpha Range}
|
| 112 |
Alpha [0.0, 1.0] controls style transfer level
|
| 113 |
--cuda Use CUDA
|
| 114 |
```
|
| 115 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 116 |
|
| 117 |
References
|
| 118 |
----------------------------
|
|
|
|
| 43 |
|
| 44 |
Download [vgg19 weight](https://drive.google.com/file/d/1UcSl-Zn3byEmn15NIPXMf9zaGCKc2gfx/view?usp=sharing), [decoder weight](https://drive.google.com/file/d/18JpLtMOapA-vwBz-LRomyTl24A9GwhTF/view?usp=sharing) under main folder.
|
| 45 |
|
| 46 |
+
To test basic style transfer, run the script test_image.py. Specify `--content_image` or `--style_img` to the image path. Specify `--content_dir` or `--style_dir` to iterate all images under this directory. Specify `--grid_pth` to collect all outputs in a grid image.
|
| 47 |
|
| 48 |
```
|
| 49 |
+
$ python test.py --content_image $IMG --style_image $STYLE --cuda
|
| 50 |
|
| 51 |
optional arguments:
|
| 52 |
-h, --help show this help message and exit
|
|
|
|
| 58 |
single style image
|
| 59 |
--style_dir STYLE_DIR
|
| 60 |
style image directory, iterate all images under this directory
|
| 61 |
+
--decoder_weight DECODER_WEIGHT decoder weight file (default='decoder.pth')
|
| 62 |
--alpha {Alpha Range}
|
| 63 |
Alpha [0.0, 1.0] controls style transfer level
|
| 64 |
--cuda Use CUDA
|
|
|
|
| 69 |
|
| 70 |
### Test Image Interpolation Style Transfer
|
| 71 |
|
| 72 |
+
To test style transfer interpolation, run the script test_interpolate.py. Specify `--style_image` with multiple paths separated by comma. Specify `--interpolation_weights` to interpolate once. Specify `--grid_pth` to interpolate with different built-in weights and provide 4 style images.
|
| 73 |
|
| 74 |
```
|
| 75 |
+
$ python test_interpolation.py --content_image $IMG --style_image $STYLE $WEIGHT --cuda
|
| 76 |
|
| 77 |
optional arguments:
|
| 78 |
-h, --help show this help message and exit
|
|
|
|
| 80 |
single content image file
|
| 81 |
--style_image STYLE_IMAGE
|
| 82 |
multiple style images file separated by comma
|
| 83 |
+
--decoder_weight DECODER_WEIGHT decoder weight file (default='decoder.pth')
|
| 84 |
--alpha {Alpha Range}
|
| 85 |
+
Alpha [0.0, 1.0] (default=1.0) controls style transfer level
|
| 86 |
--interpolation_weights INTERPOLATION_WEIGHTS
|
| 87 |
Interpolation weight of each style image, separated by comma.
|
| 88 |
Do not specify if input grid_pth.
|
|
|
|
| 99 |
To test video style transfer, run the script test_video.py.
|
| 100 |
|
| 101 |
```
|
| 102 |
+
$ python test_video.py --content_video $VID --style_image $STYLE --cuda
|
| 103 |
|
| 104 |
optional arguments:
|
| 105 |
-h, --help show this help message and exit
|
|
|
|
| 107 |
single content video file
|
| 108 |
--style_image STYLE_IMAGE
|
| 109 |
single style image
|
| 110 |
+
--decoder_weight DECODER_WEIGHT decoder weight file (default='decoder.pth')
|
| 111 |
--alpha {Alpha Range}
|
| 112 |
Alpha [0.0, 1.0] controls style transfer level
|
| 113 |
--cuda Use CUDA
|
| 114 |
```
|
| 115 |
|
| 116 |
+
Examples
|
| 117 |
+
----------------------------
|
| 118 |
+
### Test Output
|
| 119 |
+

|
| 120 |
+
|
| 121 |
+
### Different level of style transfer
|
| 122 |
+

|
| 123 |
+
|
| 124 |
+
### Interpolation Style Transfer
|
| 125 |
+

|
| 126 |
+
|
| 127 |
|
| 128 |
References
|
| 129 |
----------------------------
|