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

added simple API to SAN model #56

Closed v-goncharenko closed 4 years ago

v-goncharenko commented 4 years ago

It is able to take either file path or already loaded object (in case you load data with anything than PIL or reuse loaded images) Also it supports evaluation on both gpu and cpu

D-X-Y commented 4 years ago

Thanks a lot for your contribution.