LucasSheng / avatar-net

Avatar-Net: Multi-scale Zero-shot Style Transfer by Feature Decoration
https://lucassheng.github.io/avatar-net/
178 stars 38 forks source link

Real image size and GPU test #15

Open SolicTous opened 3 years ago

SolicTous commented 3 years ago

Hmm, nice work - really. But It's the first time when I'm not sure about real inputs.

Is my opinion is True, that the real image size, which is stylized in the network (style part) is about 512? And inputs just resized by bicubic layers before and after? Or I'm wrong? This is a fast question. I will test the model on 512 and 4k_image_size, and maybe will see the answer.

Second question. I freeze the graph and when I'm trying to infer that on GPU my kernel is dying. But CPU infer is work. It's not a good way to install old Tf, for the reason that TF needs Cuda dll's, and I have the newest CUDA and NVIDIA Drivers. (Infer is TensorFlow 2.x with Interactive session).