def stage_info_ResNet152_conv5():
return ConvStageInfo(
blobs=('res5_2_sum', 'res4_35_sum', 'res3_7_sum', 'res2_2_sum'),
dims=(2048, 1024, 512, 256),
spatial_scales=(1. / 32., 1. / 16., 1. / 8., 1. / 4.))
def stage_info_ResNet34_conv5():
return ConvStageInfo(
blobs=('res5_2_sum', 'res4_5_sum', 'res3_3_sum', 'res2_2_sum'),
dims=(512, 256, 128, 64),
spatial_scales=(1. / 32., 1. / 16., 1. / 8., 1. / 4.))