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