def __init__(self, joint_num):
super(PoseNet, self).__init__()
self.joint_num = joint_num # single hand
self.joint_deconv_1 = make_deconv_layers([2048,256,256,256])
self.joint_conv_1 = make_conv_layers([256,self.joint_num*cfg.output_hm_shape[0]],kernel=1,stride=1,padding=0,bnrelu_final=False)
self.joint_deconv_2 = make_deconv_layers([2048,256,256,256])
self.joint_conv_2 = make_conv_layers([256,self.joint_num*cfg.output_hm_shape[0]],kernel=1,stride=1,padding=0,bnrelu_final=False)
self.root_fc = make_linear_layers([2048,512,cfg.output_root_hm_shape],relu_final=False)
self.hand_fc = make_linear_layers([2048,512,2],relu_final=False)
def __init__(self, joint_num):
super(PoseNet, self).__init__()
self.joint_num = joint_num
self.deconv = make_deconv_layers([2048, 256, 256, 256])
self.conv_x = make_conv1d_layers([256, self.joint_num],
kernel=1,
stride=1,
padding=0,
bnrelu_final=False)
self.conv_y = make_conv1d_layers([256, self.joint_num],
kernel=1,
stride=1,
padding=0,
bnrelu_final=False)
self.conv_z_1 = make_conv1d_layers(
[2048, 256 * cfg.output_hm_shape[0]],
kernel=1,
stride=1,
padding=0)
self.conv_z_2 = make_conv1d_layers([256, self.joint_num],
kernel=1,
stride=1,
padding=0,
bnrelu_final=False)