search

SeungyounShin / Adaptive-Wing-Loss-for-Robust-Face-Alignment-via-Heatmap-Regression

Adaptive-Wing-Loss-for-Robust-Face-Alignment-via-Heatmap-Regression
56 stars 5 forks source link

Is it possible to create a lightweight landmark detection model using this repo? #1

Closed Keiku closed 4 years ago

Keiku commented 4 years ago

I want to make a lightweight landmark detection model like OpenSeeFace, but there is no training code in OpenSeeFace repo. Your repo has only HourglassNet, but can you make a lightweight model for mobile? Do you have any idea what is good if you make it?

SeungyounShin commented 4 years ago

DLA(deep layer aggregation) is recommended or you can stack fewer HG(hourglass) to make the model more light. however, recent approaches are focused on knowledge distillation.

Keiku commented 4 years ago

Thank you for your answer. It will be helpful.