search

tomas-gajarsky / facetorch

Python library for analysing faces using PyTorch
Apache License 2.0
503 stars 45 forks source link

Still the original size after feeding into the reader #57

Closed sun913 closed 11 months ago

sun913 commented 11 months ago

Hi!

Thanks for your amazing work first! I only need to use au and pose predictor in the model. When the image runs on the Colab demo, it will return the image with size [3,1080,1080]. However, when I only ran on my Ubuntu system, it gave me [3,288,288], which is the original size of the image. I understood that the resizing of the image is important in the later step of detecting the face of the image. I can get the face in the Colab demo but can not detect the face from my end. Can you please help me with this? Thank you!

screenshot

tomas-gajarsky commented 11 months ago

You might want to change the fix_img_size parameter to True when running the reader. Let me know if that helps.

sun913 commented 11 months ago

You might want to change the fix_img_size parameter to True when running the reader. Let me know if that helps.

Yes, that helps! Thank you.

tomas-gajarsky commented 11 months ago

I am glad it worked. In case you want to change the image size from the default 1080 pixels, you can do so by adjusting the reader configuration.