mike9251 / simswap-inference-pytorch

Unofficial Pytorch implementation (inference only) of the SimSwap: An Efficient Framework For High Fidelity Face Swapping
93 stars 20 forks source link

How to do a batch inference? #30

Open xuyingzhongguo opened 1 year ago

xuyingzhongguo commented 1 year ago

For example, the size of image here (It is not directing correctly. Should be on line 186, https://github.com/mike9251/simswap-inference-pytorch/blob/main/src/simswap.py#:~:text=detection%3A%20Detection%20%3D%20self.face_detector(image) ) is [4,3,256,256], how could we adapt the code?

Thank you in advance.