mirror of
https://github.com/huchenlei/HandRefinerPortable.git
synced 2026-03-10 12:59:47 +00:00
48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
# ------------------------------------------------------------------------------
|
|
# Copyright (c) Microsoft
|
|
# Licensed under the MIT License.
|
|
# Create by Bin Xiao (Bin.Xiao@microsoft.com)
|
|
# Modified by Ke Sun (sunk@mail.ustc.edu.cn)
|
|
# ------------------------------------------------------------------------------
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
from yacs.config import CfgNode as CN
|
|
|
|
# high_resoluton_net related params for classification
|
|
POSE_HIGH_RESOLUTION_NET = CN()
|
|
POSE_HIGH_RESOLUTION_NET.PRETRAINED_LAYERS = ['*']
|
|
POSE_HIGH_RESOLUTION_NET.STEM_INPLANES = 64
|
|
POSE_HIGH_RESOLUTION_NET.FINAL_CONV_KERNEL = 1
|
|
POSE_HIGH_RESOLUTION_NET.WITH_HEAD = True
|
|
|
|
POSE_HIGH_RESOLUTION_NET.STAGE2 = CN()
|
|
POSE_HIGH_RESOLUTION_NET.STAGE2.NUM_MODULES = 1
|
|
POSE_HIGH_RESOLUTION_NET.STAGE2.NUM_BRANCHES = 2
|
|
POSE_HIGH_RESOLUTION_NET.STAGE2.NUM_BLOCKS = [4, 4]
|
|
POSE_HIGH_RESOLUTION_NET.STAGE2.NUM_CHANNELS = [32, 64]
|
|
POSE_HIGH_RESOLUTION_NET.STAGE2.BLOCK = 'BASIC'
|
|
POSE_HIGH_RESOLUTION_NET.STAGE2.FUSE_METHOD = 'SUM'
|
|
|
|
POSE_HIGH_RESOLUTION_NET.STAGE3 = CN()
|
|
POSE_HIGH_RESOLUTION_NET.STAGE3.NUM_MODULES = 1
|
|
POSE_HIGH_RESOLUTION_NET.STAGE3.NUM_BRANCHES = 3
|
|
POSE_HIGH_RESOLUTION_NET.STAGE3.NUM_BLOCKS = [4, 4, 4]
|
|
POSE_HIGH_RESOLUTION_NET.STAGE3.NUM_CHANNELS = [32, 64, 128]
|
|
POSE_HIGH_RESOLUTION_NET.STAGE3.BLOCK = 'BASIC'
|
|
POSE_HIGH_RESOLUTION_NET.STAGE3.FUSE_METHOD = 'SUM'
|
|
|
|
POSE_HIGH_RESOLUTION_NET.STAGE4 = CN()
|
|
POSE_HIGH_RESOLUTION_NET.STAGE4.NUM_MODULES = 1
|
|
POSE_HIGH_RESOLUTION_NET.STAGE4.NUM_BRANCHES = 4
|
|
POSE_HIGH_RESOLUTION_NET.STAGE4.NUM_BLOCKS = [4, 4, 4, 4]
|
|
POSE_HIGH_RESOLUTION_NET.STAGE4.NUM_CHANNELS = [32, 64, 128, 256]
|
|
POSE_HIGH_RESOLUTION_NET.STAGE4.BLOCK = 'BASIC'
|
|
POSE_HIGH_RESOLUTION_NET.STAGE4.FUSE_METHOD = 'SUM'
|
|
|
|
MODEL_EXTRAS = {
|
|
'cls_hrnet': POSE_HIGH_RESOLUTION_NET,
|
|
}
|