|
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