lengstrom / fast-style-transfer

TensorFlow CNN for fast style transfer ⚡🖥🎨🖼
10.91k stars 2.6k forks source link

how to change 'resize' resolution? #271

Open zuozhen opened 3 years ago

zuozhen commented 3 years ago

I read the code and I got the training run successfully. But after searching the project I did not find where to config param of "resize the image" (e.g. I'd like resize image to 800 x 800, for a better-performance model), please give help if anyone knows, thanks very much.

By the way, I have two other questions:

  1. Line 161 of style.py says that "save_img(preds_path, img)", but the param "img" is not defined, how could that work without error?
  2. Line 14 of "src/utils.py" says that "style_target = _get_img(style_path, img_size=new_shape)". There is no "_get_img" method defined, and even without import it. Why no error occur when running this?

Thanks.