From ae6cb7d529430f5c4a5d6246e8ca121ebf2e1a45 Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Mon, 15 Nov 2021 02:54:39 +0000
Subject: [PATCH] feat: support MixNet

---
 paddleclas.py                   | 1 +
 ppcls/arch/backbone/__init__.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/paddleclas.py b/paddleclas.py
index 58a3cddd..6ccdbd0b 100644
--- a/paddleclas.py
+++ b/paddleclas.py
@@ -73,6 +73,7 @@ MODEL_SERIES = {
         "HRNet_W48_C_ssld"
     ],
     "Inception": ["GoogLeNet", "InceptionV3", "InceptionV4"],
+    "MixNet": ["MixNet_S", "MixNet_M", "MixNet_L"],
     "MobileNetV1": [
         "MobileNetV1_x0_25", "MobileNetV1_x0_5", "MobileNetV1_x0_75",
         "MobileNetV1", "MobileNetV1_ssld"
diff --git a/ppcls/arch/backbone/__init__.py b/ppcls/arch/backbone/__init__.py
index 1764830d..aa95ebc7 100644
--- a/ppcls/arch/backbone/__init__.py
+++ b/ppcls/arch/backbone/__init__.py
@@ -65,6 +65,7 @@ from ppcls.arch.backbone.variant_models.vgg_variant import VGG19Sigmoid
 from ppcls.arch.backbone.variant_models.pp_lcnet_variant import PPLCNet_x2_5_Tanh
 
 
+# help whl get all the models' api (class type) and components' api (func type)
 def get_apis():
     current_func = sys._getframe().f_code.co_name
     current_module = sys.modules[__name__]
-- 
GitLab