diff options
| author | cam <cameron@ideum.com> | 2016-10-17 02:23:30 -0600 |
|---|---|---|
| committer | cam <cameron@ideum.com> | 2016-10-17 02:23:30 -0600 |
| commit | 2e1f05fe478a5797f991b4b4fbfdac82726f210a (patch) | |
| tree | 0e91e995b359c3a64d9f7384f20f1bf681575bea | |
| parent | 5ca01d831a0ceec853135b9eb35e65ed8e327d82 (diff) | |
Added multiple content layers
| -rw-r--r-- | neural_style.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/neural_style.py b/neural_style.py index 1dd0c6c..4c08220 100644 --- a/neural_style.py +++ b/neural_style.py @@ -72,7 +72,7 @@ def parse_args(): choices=[1, 2, 3], help='Different constants for the content layer loss function. (default: %(default)s)') - parser.add_argument('--content_layers', type=str, + parser.add_argument('--content_layers', nargs='+', type=str, default=['conv4_2'], help='VGG19 layers used for the content image. (default: %(default)s)') |