From 2e1f05fe478a5797f991b4b4fbfdac82726f210a Mon Sep 17 00:00:00 2001 From: cam Date: Mon, 17 Oct 2016 02:23:30 -0600 Subject: Added multiple content layers --- neural_style.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)') -- cgit v1.2.3-70-g09d2