From 5ff77e27cafffa6e74e8b8015415828819014669 Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Wed, 8 Jun 2022 14:32:56 +0000
Subject: [PATCH] feat: add pplcnetv2 pphgnet

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

diff --git a/paddleclas.py b/paddleclas.py
index affc4833..ef1c47da 100644
--- a/paddleclas.py
+++ b/paddleclas.py
@@ -100,10 +100,17 @@ IMN_MODEL_SERIES = {
         "MobileNetV3_large_x1_0", "MobileNetV3_large_x1_25",
         "MobileNetV3_small_x1_0_ssld", "MobileNetV3_large_x1_0_ssld"
     ],
+    "PPHGNet": [
+        "PPHGNet_tiny",
+        "PPHGNet_small",
+        "PPHGNet_tiny_ssld",
+        "PPHGNet_small_ssld",
+    ],
     "PPLCNet": [
         "PPLCNet_x0_25", "PPLCNet_x0_35", "PPLCNet_x0_5", "PPLCNet_x0_75",
         "PPLCNet_x1_0", "PPLCNet_x1_5", "PPLCNet_x2_0", "PPLCNet_x2_5"
     ],
+    "PPLCNetV2": ["PPLCNetV2_base"],
     "RedNet": ["RedNet26", "RedNet38", "RedNet50", "RedNet101", "RedNet152"],
     "RegNet": ["RegNetX_4GF"],
     "Res2Net": [
-- 
GitLab