print(ResNet3dPathway(**pose_pathway).lateral)
print(ResNet3dPathway(**pose_pathway).lateral_activate[0]==1)
print(ResNet3dPathway(**pose_pathway).stage_blocks)
print(len(ResNet3dPathway(**pose_pathway).stage_blocks))
print(ResNet3dPathway(**pose_pathway).base_channels * 2**0)
print(ResNet3dPathway(**pose_pathway).block.expansion)
print(ResNet3dPathway(**pose_pathway).lateral_activate[1])
print(ResNet3dPathway(**pose_pathway).lateral_inv)
print(ResNet3dPathway(**pose_pathway).channel_ratio)
print(ResNet3dPathway(**pose_pathway).lateral_infl)
print(ResNet3dPathway(**pose_pathway).lateral_connections)
True
False
(4, 6, 3)
3
32
4
1
True
4
16
['layer1_lateral', 'layer2_lateral']
print(ResNet3dPathway(**pose_pathway))
(layer1_lateral): DeConvModule(
(conv): ConvTranspose3d(512, 32, kernel_size=(7, 1, 1), stride=(4, 1, 1), padding=(3, 0, 0), bias=False)
(bn): BatchNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(relu): ReLU()
)
(layer2_lateral): DeConvModule(
(conv): ConvTranspose3d(1024, 64, kernel_size=(7, 1, 1), stride=(4, 1, 1), padding=(3, 0, 0), bias=False)
(bn): BatchNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(relu): ReLU()
)
经过比对发现参数和自己手动计算一致
https://github.com/kennymckormick/pyskl/blob/main/pyskl/models/cnns/resnet3d_slowfast.py