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 #78

Open nagadit opened 3 years ago

nagadit commented 3 years ago

Hi!!! How I can inference pretrained SRT model ? You have script like eval.py from SBR?

D-X-Y commented 3 years ago

@nagadit I just updated the evaluation of SRT codes. You can have a try of this command: https://github.com/D-X-Y/landmark-detection/blob/master/SRT/exps/BASE-eval-image.py#L8

nagadit commented 3 years ago

Thanks a lot! Pre-trained model to use this? https://drive.google.com/drive/folders/14oSC646HRtEI1H14gq1KVOkE2Tzs051h

D-X-Y commented 3 years ago

Yeap~

flynnamy commented 3 years ago

@nagadit Hi,could you do inference and get right results?I can do inference but get wrong results like following pic. image

nagadit commented 3 years ago

@flynnamy why do you think this is a wrong result?

flynnamy commented 3 years ago

It is the coordinates of X,Y. This results can not show in the image. @nagadit

nagadit commented 3 years ago

@flynnamy draw rectangle on your img and show pls)

D-X-Y commented 3 years ago

Hi all, please refer to this problem (https://github.com/D-X-Y/landmark-detection/issues/81) I just fixed the SRT single image evaluation issue (This is the new file https://github.com/D-X-Y/landmark-detection/blob/master/SRT/exps/BASE-eval-image.py)

Sorry for the inconvenience.