def resnet_v2_block(scope, base_depth, num_units, stride):
return utils.Block(scope, block, [{
'depth': base_depth * 4,
'stride': stride
}] + (num_units - 1) * [{
'depth': base_depth * 4,
'stride': 1
}])
def resnet_v1_block(scope, base_depth, num_units, stride):
return utils.Block(scope, bottleneck, [{
'depth': base_depth * 4,
'depth_bottleneck': base_depth,
'stride': 1
}] * (num_units - 1) + [{
'depth': base_depth * 4,
'depth_bottleneck': base_depth,
'stride': stride
}])