|
21 | 21 | POSE_RESNET.FINAL_CONV_KERNEL = 1
|
22 | 22 | POSE_RESNET.PRETRAINED_LAYERS = ['*']
|
23 | 23 |
|
24 |
| -# pose_resnet related params |
25 |
| -POSE_RESNEXT = CN() |
26 |
| -POSE_RESNEXT.NUM_LAYERS = 50 |
27 |
| -POSE_RESNEXT.DECONV_WITH_BIAS = False |
28 |
| -POSE_RESNEXT.NUM_DECONV_LAYERS = 3 |
29 |
| -POSE_RESNEXT.NUM_DECONV_FILTERS = [256, 256, 256] |
30 |
| -POSE_RESNEXT.NUM_DECONV_KERNELS = [4, 4, 4] |
31 |
| -POSE_RESNEXT.FINAL_CONV_KERNEL = 1 |
32 |
| -POSE_RESNEXT.BASE_WIDTH = 4 |
33 |
| -POSE_RESNEXT.CARDINALITY = 32 |
34 |
| -POSE_RESNEXT.STEM_KERNEL_SIZE = 7 |
35 |
| -POSE_RESNEXT.PRETRAINED_LAYERS = ['*'] |
36 |
| - |
37 | 24 | # pose_multi_resoluton_net related params
|
38 | 25 | POSE_HIGH_RESOLUTION_NET = CN()
|
39 | 26 | POSE_HIGH_RESOLUTION_NET.PRETRAINED_LAYERS = ['*']
|
40 | 27 | POSE_HIGH_RESOLUTION_NET.STEM_INPLANES = 64
|
41 | 28 | POSE_HIGH_RESOLUTION_NET.FINAL_CONV_KERNEL = 1
|
42 | 29 |
|
43 |
| -# POSE_HIGH_RESOLUTION_NET.STAGE1 = CN() |
44 |
| -# POSE_HIGH_RESOLUTION_NET.STAGE1.NUM_MODULES = 1 |
45 |
| -# POSE_HIGH_RESOLUTION_NET.STAGE1.NUM_BRANCHES = 1 |
46 |
| -# POSE_HIGH_RESOLUTION_NET.STAGE1.NUM_BLOCKS = [4] |
47 |
| -# POSE_HIGH_RESOLUTION_NET.STAGE1.NUM_CHANNELS = [64] |
48 |
| -# POSE_HIGH_RESOLUTION_NET.STAGE1.BLOCK = 'BOTTLENECK' |
49 |
| -# POSE_HIGH_RESOLUTION_NET.STAGE1.FUSE_METHOD = 'CAT' |
50 |
| - |
51 | 30 | POSE_HIGH_RESOLUTION_NET.STAGE2 = CN()
|
52 | 31 | POSE_HIGH_RESOLUTION_NET.STAGE2.NUM_MODULES = 1
|
53 | 32 | POSE_HIGH_RESOLUTION_NET.STAGE2.NUM_BRANCHES = 2
|
|
75 | 54 |
|
76 | 55 | MODEL_EXTRAS = {
|
77 | 56 | 'pose_resnet': POSE_RESNET,
|
78 |
| - 'pose_resnetstem3x3': POSE_RESNET, |
79 |
| - 'pose_resNeXt': POSE_RESNEXT, |
80 | 57 | 'pose_high_resolution_net': POSE_HIGH_RESOLUTION_NET,
|
81 | 58 | }
|
0 commit comments