search

mantasu / face-crop-plus

Face aligner and cropper with quality enhancement and attribute parsing
https://mantasu.github.io/face-crop-plus/
MIT License
53 stars 6 forks source link

the link of pretrained model #7

Open JavanehBahrami opened 6 months ago

JavanehBahrami commented 6 months ago

could you please, add the link to the pertained model for predicting facial attributes?

mantasu commented 6 months ago

Hi, it's the same model as for the segmentation, which is an asset of the initial release. It is automatically downloaded during attribute identification but you can download it manually too: bise_parser.pth. Attribute detection is controlled via attr_threshold.

It is just an extension of the segmentation model, thus further attributes are not considered. Perhaps this could be considered in the future release but it's hard to account for many different facial attributes unless there is a dataset with some common/most important ones.