コード例 #1
0
ファイル: modifiedResNet_v2.py プロジェクト: minha12/APF
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
    }])
コード例 #2
0
ファイル: ResNet_v1.py プロジェクト: minha12/APF
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
    }])