/research

Primary LanguageJupyter Notebook

research

Automatic Kinship Recognition

My Aims

  • Adapt Shadrikov code to PyTorch.
  • Reproduce Shadrikov verification results.
    • Pretrain ResNet101 with ArcFace loss on cleaned MS-Celeb-1M.
    • Finetune pretrained model
      • pretrain + classification
      • pretrain + classification + normalization
      • pretrain + classification + normalization + different thresholds
  • Experiment with transformers instead of the ResNet backbone on both MS-Celeb-1M pretraining and FIW finetuning.

Results

Specs

  • Intel(R) Core(TM) i9-9900K CPU @ 3.60GHz
  • 48GB RAM
  • 2x NVIDIA GeForce RTX 3090 24GB

CLI

Pretrain

TODO

Metrics

TODO