research

Automatic Kinship Recognition

MXNet original implementation: vuvko/fitw2020

My Aims

Adapt Shadrikov code to PyTorch.
Reproduce Shadrikov verification results.
- Pretrain ResNet101 with ArcFace loss on cleaned MS-Celeb-1M.
  - Results here.
- 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