search

HelloVision / HelloMeme

The official HelloMeme GitHub site
https://songkey.github.io/hellomeme/
MIT License
199 stars 14 forks source link

Error in image alignment preprocessing #10

Open laxyon opened 2 weeks ago

laxyon commented 2 weeks ago

thanks for your interesting work! i wanna try it locally, but got an issuse in the preprocessing https://github.com/HelloVision/HelloMeme/blob/7dd30d97f0f9590d077e84a668df9901d23da713/hellomeme/tools/hello_face_alignment.py#L30

this input_onnx size is not match to your "face_alignment_net_222" input_shape

onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: input.1 for the following indices index: 2 Got: 128 Expected: 272 index: 3 Got: 128 Expected: 160

songkey commented 2 weeks ago

This appears to be an border case related to the input. I haven’t been able to reproduce it on my end yet. If possible, could you provide the test sample (songkey@pku.edu.cn)? That way, I can locate and fix the issue right away.