diff --git a/paddleclas.py b/paddleclas.py index 58a3cddd3c674c0114ce8c8eec71b61483f7f3c9..6ccdbd0bae17465f0ae696d8f012dec81e23df75 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 1764830dcb01262d2c60dfd2ab7a2c8dc6670324..aa95ebc7806db6e17b6f2f8a3785c058f60e77f0 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__]