def cbam(shape=(img_height, img_width, img_channel)):
depth = 50
attention_module = 'cbam_block'
model = resnet_v1(input_shape=shape,
depth=depth,
num_classes=num_classes,
attention_module=attention_module)
return model
def resnet152_se(shape=(img_height, img_width, img_channel)):
depth = 152
attention_module = 'se_block'
model = resnet_v1(input_shape=shape,
depth=depth,
num_classes=num_classes,
attention_module=attention_module)
return model