falseu commited on
Commit
341b820
·
1 Parent(s): e4050d7

update readme

Browse files
Files changed (1) hide show
  1. README.md +20 -9
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 `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 --decoder_weight $WEIGHT --cuda
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 `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 --decoder_weight $WEIGHT --cuda
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 --decoder_weight $WEIGHT --cuda
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
+ ![grid](https://github.com/media-comp/2022-AdaIN-pytorch/tree/main/examples/grid.jpg)
120
+
121
+ ### Different level of style transfer
122
+ ![grid_alpha](https://github.com/media-comp/2022-AdaIN-pytorch/tree/main/examples/grid_alpha.png)
123
+
124
+ ### Interpolation Style Transfer
125
+ ![grid_inter](https://github.com/media-comp/2022-AdaIN-pytorch/tree/main/examples/grid_interpolation.png)
126
+
127
 
128
  References
129
  ----------------------------