提交 b07e6612 编写于 作者: G gaotingquan 提交者: Tingquan Gao

feat: support ConvNeXt, VAN_B0, CSPDarkNet53, PeleeNet

上级 e21c060f
...@@ -51,6 +51,8 @@ BASE_IMAGES_DIR = os.path.join(BASE_DIR, "images") ...@@ -51,6 +51,8 @@ BASE_IMAGES_DIR = os.path.join(BASE_DIR, "images")
IMN_MODEL_BASE_DOWNLOAD_URL = "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/inference/{}_infer.tar" IMN_MODEL_BASE_DOWNLOAD_URL = "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/inference/{}_infer.tar"
IMN_MODEL_SERIES = { IMN_MODEL_SERIES = {
"AlexNet": ["AlexNet"], "AlexNet": ["AlexNet"],
"ConvNeXt": ["ConvNeXt_tiny"],
"CSPNet": ["CSPDarkNet53"],
"CSWinTransformer": [ "CSWinTransformer": [
"CSWinTransformer_tiny_224", "CSWinTransformer_small_224", "CSWinTransformer_tiny_224", "CSWinTransformer_small_224",
"CSWinTransformer_base_224", "CSWinTransformer_base_384", "CSWinTransformer_base_224", "CSWinTransformer_base_384",
...@@ -108,6 +110,7 @@ IMN_MODEL_SERIES = { ...@@ -108,6 +110,7 @@ IMN_MODEL_SERIES = {
"MobileNetV3_small_x1_0_ssld", "MobileNetV3_large_x1_0_ssld" "MobileNetV3_small_x1_0_ssld", "MobileNetV3_large_x1_0_ssld"
], ],
"MobileViT": ["MobileViT_XXS", "MobileViT_XS", "MobileViT_S"], "MobileViT": ["MobileViT_XXS", "MobileViT_XS", "MobileViT_S"],
"PeleeNet": ["PeleeNet"],
"PPHGNet": [ "PPHGNet": [
"PPHGNet_tiny", "PPHGNet_tiny",
"PPHGNet_small", "PPHGNet_small",
...@@ -176,6 +179,7 @@ IMN_MODEL_SERIES = { ...@@ -176,6 +179,7 @@ IMN_MODEL_SERIES = {
"alt_gvt_base", "alt_gvt_large" "alt_gvt_base", "alt_gvt_large"
], ],
"TNT": ["TNT_small"], "TNT": ["TNT_small"],
"VAN": ["VAN_B0"],
"VGG": ["VGG11", "VGG13", "VGG16", "VGG19"], "VGG": ["VGG11", "VGG13", "VGG16", "VGG19"],
"VisionTransformer": [ "VisionTransformer": [
"ViT_base_patch16_224", "ViT_base_patch16_384", "ViT_base_patch32_384", "ViT_base_patch16_224", "ViT_base_patch16_384", "ViT_base_patch32_384",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册