Skip to content

Commit ca0cb0b

Browse files
committed
Added pretrained weights for soft classification and ranking SFCN
1 parent c545dd7 commit ca0cb0b

1 file changed

Lines changed: 20 additions & 0 deletions

File tree

pyment/models/utils/weight_repository.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,26 @@
1313
'url': ('https://drive.google.com/u/0/'
1414
'uc?id=1Mev57Rdst5TAxGEt3OUY97ToKOWq0uCC&export=download'),
1515
'filename': 'regression_sfcn_brain_age_weights_no_top.hdf5'
16+
},
17+
('SoftClassificationSFCN', 'brain-age', True): {
18+
'url': ('https://drive.google.com/u/0/'
19+
'uc?id=1_sQu4HoLUIHXcL0qO9sHcsKnH-WCVGDK&export=download'),
20+
'filename': 'soft_classification_sfcn_brain_age_weights.hdf5'
21+
},
22+
('SoftClassificationSFCN', 'brain-age', False): {
23+
'url': ('https://drive.google.com/u/0/'
24+
'uc?id=1gO-LW96lQVCMp6mZ_rfbTkrgiHacgIoS&export=download'),
25+
'filename': 'soft_classification_sfcn_brain_age_weights_no_top.hdf5'
26+
},
27+
('RankingSFCN', 'brain-age', True): {
28+
'url': ('https://drive.google.com/u/0/'
29+
'uc?id=1cElx0AufQWDAKUWoELFQ4piBbgE4AIAz&export=download'),
30+
'filename': 'ranking_sfcn_brain_age_weights.hdf5'
31+
},
32+
('RankingSFCN', 'brain-age', False): {
33+
'url': ('https://drive.google.com/u/0/'
34+
'uc?id=1-2s2psZD2R9-ReTOHHJLzJZeYJAxQGBK&export=download'),
35+
'filename': 'ranking_sfcn_brain_age_weights_no_top.hdf5'
1636
}
1737
}
1838

0 commit comments

Comments
 (0)