Skip to content

Latest commit

 

History

History
23 lines (17 loc) · 688 Bytes

File metadata and controls

23 lines (17 loc) · 688 Bytes

A pytorch reimplementation of SSR-Net by

author : oukohou
time : 2019-09-26 16:44:48
email : oukohou@outlook.com

This code is forked from here

Dataset

To finetune SSR-Net on your data, prepare train/val/test annotation files in the following format:

img_name,age,age_group
26_0_4_20170117195456708.jpg.chip.jpg,26,(20;29)
34_0_1_20170116002355232.jpg.chip.jpg,34,(30;39)

Replace the annotation file paths in SSR_NET_FT/train_SSR-Net.py and update the base_path to your dataset root.

Fine-tuning

python train_SSR-Net.py