From b07e661294ababc1f65d48c205305bfdc14b0e88 Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Tue, 18 Oct 2022 08:50:13 +0000
Subject: [PATCH] feat: support ConvNeXt, VAN_B0, CSPDarkNet53, PeleeNet

---
 paddleclas.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/paddleclas.py b/paddleclas.py
index 1f107af4f..5022ad8fb 100644
--- a/paddleclas.py
+++ b/paddleclas.py
@@ -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_SERIES = {
     "AlexNet": ["AlexNet"],
+    "ConvNeXt": ["ConvNeXt_tiny"],
+    "CSPNet": ["CSPDarkNet53"],
     "CSWinTransformer": [
         "CSWinTransformer_tiny_224", "CSWinTransformer_small_224",
         "CSWinTransformer_base_224", "CSWinTransformer_base_384",
@@ -108,6 +110,7 @@ IMN_MODEL_SERIES = {
         "MobileNetV3_small_x1_0_ssld", "MobileNetV3_large_x1_0_ssld"
     ],
     "MobileViT": ["MobileViT_XXS", "MobileViT_XS", "MobileViT_S"],
+    "PeleeNet": ["PeleeNet"],
     "PPHGNet": [
         "PPHGNet_tiny",
         "PPHGNet_small",
@@ -176,6 +179,7 @@ IMN_MODEL_SERIES = {
         "alt_gvt_base", "alt_gvt_large"
     ],
     "TNT": ["TNT_small"],
+    "VAN": ["VAN_B0"],
     "VGG": ["VGG11", "VGG13", "VGG16", "VGG19"],
     "VisionTransformer": [
         "ViT_base_patch16_224", "ViT_base_patch16_384", "ViT_base_patch32_384",