fast-style-transfer icon indicating copy to clipboard operation
fast-style-transfer copied to clipboard

how to change 'resize' resolution?

Open zuozhen opened this issue 3 years ago • 0 comments

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.

zuozhen avatar May 20 '21 08:05 zuozhen