neural-style-pt icon indicating copy to clipboard operation
neural-style-pt copied to clipboard

Can you share your training parameters? I can't get as good a performance as you.

Open yunlm opened this issue 3 years ago • 4 comments

yunlm avatar Jul 26 '22 14:07 yunlm

@yunlm Try using the following parameters with multiscale generation:

-init image -tv_weight 0 -learning_rate 2 -init image -content_weight 100 -style_weight 6000

ProGamerGov avatar Jul 26 '22 15:07 ProGamerGov

maroon

Starry_Night

Using the above two images and the following parameters `python neural_style.py -style_image ~/Documents/NST/style/Starry-Night.jpg -content_image ~/Documents/NST/maroon.JPG -output_image maroonxstarry.jpg -model_file models/vgg19-d01eb7cb.pth -gpu 0 -num_iterations 2000 -content_weight 10 -style_weight 50000 -image_size 1024 -optimizer lbfgs -init image -tv_weight 0 -learning_rate 2 -init image -style_scale 1.5 -original_colors 1', I produced this:

MaroonXStarry

I'm finding the style_scale parameter to be very useful. Here's the same pair redone using a style scale of 0.5 MaroonXStarry_scale0 5 :

strohrbaugh avatar Aug 15 '22 04:08 strohrbaugh

@strohrbaugh You should start off at a lower image size and then scale it up with multiple steps of increasing image size (multiscale rendering).

ProGamerGov avatar Aug 15 '22 13:08 ProGamerGov

You're so right. I'm glad I finally made the time to try that out.

strohrbaugh avatar Dec 02 '22 22:12 strohrbaugh