mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add ViT and DeiT ()

pull/670/head
Tingquan Gao 2021-04-06 15:12:26 +08:00 committed by GitHub
parent 4523d4246d
commit 794caef23b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 3 deletions

12
paddleclas.py
View File

@ -86,7 +86,13 @@ model_names = {
'InceptionV3', 'ShuffleNetV2_swish', 'GoogLeNet', 'ResNet50_vd_ssld_v2',
'SE_ResNet50_vd', 'MobileNetV2', 'ResNeXt101_vd_32x4d',
'MobileNetV3_large_x0_75', 'MobileNetV3_small_x0_5', 'DenseNet169',
'EfficientNetB5'
'EfficientNetB5', 'DeiT_base_distilled_patch16_224',
'DeiT_base_distilled_patch16_384', 'DeiT_base_patch16_224',
'DeiT_base_patch16_384_infer', 'DeiT_small_distilled_patch16_224',
'DeiT_small_patch16_224_infer', 'DeiT_tiny_distilled_patch16_224',
'DeiT_tiny_patch16_224', 'ViT_base_patch16_224', 'ViT_base_patch16_384',
'ViT_base_patch32_384', 'ViT_large_patch16_224', 'ViT_large_patch16_384',
'ViT_large_patch32_384', 'ViT_small_patch16_224'
}
@ -253,8 +259,8 @@ class PaddleClas(object):
__dir__, 'ppcls/utils/imagenet1k_label_list.txt')
else:
raise Exception(
'If you want to use your own model, Please input model_file as model path!'
)
'The model inputed is {}, not provided by PaddleClas. If you want to use your own model, please input model_file as model path!'.
format(process_params.model_name))
else:
print('Using user-specified model and params!')