search

sunshineatnoon / Paper-Implementations

Use PyTorch to implement some classic frameworks
620 stars 197 forks source link

util.open_and_resize_image should not be hard code #7

Open ttpro1995 opened 4 years ago

ttpro1995 commented 4 years ago

In script NeuralSytleTransfer/train.py

the open_and_resize function should not hard-code 

styleImg = transform(util.open_and_resize_image(opt.style_image,256))
contentImg = transform(util.open_and_resize_image(opt.content_image,256))

rather, it should use parameters from opt.imageSize