search

D-X-Y / landmark-detection

Four landmark detection algorithms, implemented in PyTorch.
https://xuanyidong.com/assets/projects/TPAMI-2020-SRT.html
MIT License
925 stars 180 forks source link

SRT inference get wrong results #81

Closed flynnamy closed 3 years ago

flynnamy commented 3 years ago

Hi, I execute this command:python exps/BASE-eval-image.py --image ./cache_data/cache/self.jpeg --face 250 150 900 1100 --model checkpoint-regression.pth,but get wrong landmark like followings pic. Any suggestions? @D-X-Y image

AndriiHura commented 3 years ago

I can't even run inference with this --model checkpoint-regression.pth checkpoint as I right away get a Missing key(s) in state_dict: error.

D-X-Y commented 3 years ago

@AndriiHura Could you please show your step-by-step commands? The missing keys are due to the mismatch between DaraParallel Model and the original model.

Would you mind to re-try? I updated the codes to fix this issue.

flynnamy commented 3 years ago

@D-X-Y Please see my results, Any suggestions?

D-X-Y commented 3 years ago

@flynnamy It was my fault that used an old eval file from SBR.. I just updated the codes, please have a try now. I tried it on my laptop and works fine.

D-X-Y commented 3 years ago

Please feel free to re-open this issue if you have another question.

AndriiHura commented 3 years ago

@D-X-Y Now it works great! Thanks a lot :)