Upload directory
Browse files
aligners/differentiable_face_aligner/dfa/config.py
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# config.py
|
| 2 |
+
|
| 3 |
+
cfg_mnet = {
|
| 4 |
+
'name': 'mobilenet0.25',
|
| 5 |
+
'pretrain': True,
|
| 6 |
+
'return_layers': {'stage1': 1, 'stage2': 2, 'stage3': 3},
|
| 7 |
+
'in_channel': 32,
|
| 8 |
+
'out_channel': 64
|
| 9 |
+
}
|
| 10 |
+
|
| 11 |
+
cfg_re50 = {
|
| 12 |
+
'name': 'Resnet50',
|
| 13 |
+
'pretrain': True,
|
| 14 |
+
'return_layers': {'layer2': 1, 'layer3': 2, 'layer4': 3},
|
| 15 |
+
'in_channel': 256,
|
| 16 |
+
'out_channel': 256
|
| 17 |
+
}
|
| 18 |
+
|