对于检测算法,尤其是两阶段的检测算法,通常需要使用预训练模型来初始化网络参数,才能够收敛到比较好的效果;这里提供了常见模型的预训练模型。

model name type
vgg16_reduced_fc detection
vgg16 all
resnet_v1_50 all
resnet_v1_101 all
inception_v1 all
inception_v2 all
inception_v3 all
inception_v4 all
mobilenet_v1_1.0_224 all
mobilenet_v2_1.0_224 all
xception_41 all
xception_65 all
xception_71 all
se_resnet_v1_50 all
se_resnet_v1_101 all
se_resnet_v1_152 all
synthtext_text_end2end_krcnn_attn_resnet50 text_end2end
general_text_end2end_krcnn_attn_resnet50 text_end2end
general_crnn_attn_resnet15 text_recognition

视频pretrain模型

model_name type
resnet3d_50 video_classification
resnet3d_101 video_classification