Skip to content

Implementation of a classification model that incorporates arcface loss and GeM pooling to improve intra class closeness and inter class seperability

License

Notifications You must be signed in to change notification settings

LawJarp-A/arcface-classification-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

arcface-classification-pytorch

Implementation of a classification model that incorporates arcface loss and GeM pooling to improve intra class closeness and inter class seperability. PyTorch is primarily used along with fastai framework to speed up development and benchmarking.

Arcface loss

Refer to this paper for more information

How to set the parameters for Arcface loss

  • There are a few hyper parameters that are dependent on the number of classes and how much error one can sustain.
  • Refer this wonderful article about using arcface and it's components and how to choose the right values for the hyper parameters

About

Implementation of a classification model that incorporates arcface loss and GeM pooling to improve intra class closeness and inter class seperability

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages