changed README.md

README.md

@@ -44,7 +44,7 @@ optional arguments:
 
 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 the main directory.
 
-To test basic style transfer, run the script test.py. Specify `--content_image`, `--style_image` to the image path, or specify `--content_dir`, `--style_dir` to iterate all images under this directory. All outputs are saved in `./results/`. Specify `--grid_pth` to collect all outputs in a grid image.
+To test basic style transfer, run the script test.py. Specify `--content_image`, `--style_image` to the image path, or specify `--content_dir`, `--style_dir` to iterate all images under this directory. All outputs are saved in `./results/`. Specify `--grid_pth` to collect all outputs in a grid image. Specify `--color_control` to preserve the content image color.
 
 ```
 $ python test.py --content_image $IMG --style_image $STYLE --cuda
@@ -66,12 +66,12 @@ optional arguments:
   --grid_pth GRID_PTH
                         Specify a grid image path (default=None) if generate a grid image
                         that contains all style transferred images
-  --color_control       Preserve content color
+  --color_control       Preserve content image color
 ```
 
 ### Test Image Interpolation Style Transfer
 
-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. All outputs are saved in `./results_interpolate/`. Specify `--grid_pth` to interpolate with different built-in weights and provide 4 style images.
+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. All outputs are saved in `./results_interpolate/`. Specify `--grid_pth` to interpolate with different built-in weights and provide 4 style images. Specify `--color_control` to preserve the content image color.
 
 ```
 
@@ -96,7 +96,7 @@ optional arguments:
                         transfer multiple times with different built-in weights and generate a
                         grid image that contains all style transferred images. Provide 4 style
                         images. Do not specify if input interpolation_weights.
-  --color_control       Preserve content color
+  --color_control       Preserve content image color
 ```
 
 ### Test Video Style Transfer
@@ -116,7 +116,7 @@ optional arguments:
   --alpha {Alpha Range}
                         Alpha [0.0, 1.0] controls style transfer level
   --cuda                Use CUDA
-  --color_control       Preserve content color
+  --color_control       Preserve content image color
 ```
 
 ## Examples