Skip to content

Commit b07e661

Browse files
committed
feat: support ConvNeXt, VAN_B0, CSPDarkNet53, PeleeNet
1 parent e21c060 commit b07e661

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

paddleclas.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,8 @@
5151
IMN_MODEL_BASE_DOWNLOAD_URL = "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/inference/{}_infer.tar"
5252
IMN_MODEL_SERIES = {
5353
"AlexNet": ["AlexNet"],
54+
"ConvNeXt": ["ConvNeXt_tiny"],
55+
"CSPNet": ["CSPDarkNet53"],
5456
"CSWinTransformer": [
5557
"CSWinTransformer_tiny_224", "CSWinTransformer_small_224",
5658
"CSWinTransformer_base_224", "CSWinTransformer_base_384",
@@ -108,6 +110,7 @@
108110
"MobileNetV3_small_x1_0_ssld", "MobileNetV3_large_x1_0_ssld"
109111
],
110112
"MobileViT": ["MobileViT_XXS", "MobileViT_XS", "MobileViT_S"],
113+
"PeleeNet": ["PeleeNet"],
111114
"PPHGNet": [
112115
"PPHGNet_tiny",
113116
"PPHGNet_small",
@@ -176,6 +179,7 @@
176179
"alt_gvt_base", "alt_gvt_large"
177180
],
178181
"TNT": ["TNT_small"],
182+
"VAN": ["VAN_B0"],
179183
"VGG": ["VGG11", "VGG13", "VGG16", "VGG19"],
180184
"VisionTransformer": [
181185
"ViT_base_patch16_224", "ViT_base_patch16_384", "ViT_base_patch32_384",

0 commit comments

Comments
 (0)