提交 be37ba87 编写于 作者: G gaotingquan

feat: support Twins & PPLCNet

上级 6e4bf593
...@@ -14,3 +14,4 @@ ...@@ -14,3 +14,4 @@
__all__ = ['PaddleClas'] __all__ = ['PaddleClas']
from .paddleclas import PaddleClas from .paddleclas import PaddleClas
from ppcls.arch.backbone import *
...@@ -38,6 +38,7 @@ from deploy.utils.get_image_list import get_image_list ...@@ -38,6 +38,7 @@ from deploy.utils.get_image_list import get_image_list
from deploy.utils import config from deploy.utils import config
from ppcls.arch.backbone import * from ppcls.arch.backbone import *
from ppcls.utils.logger import init_logger
__all__ = ["PaddleClas"] __all__ = ["PaddleClas"]
...@@ -89,6 +90,10 @@ MODEL_SERIES = { ...@@ -89,6 +90,10 @@ MODEL_SERIES = {
"MobileNetV3_large_x1_0", "MobileNetV3_large_x1_25", "MobileNetV3_large_x1_0", "MobileNetV3_large_x1_25",
"MobileNetV3_small_x1_0_ssld", "MobileNetV3_large_x1_0_ssld" "MobileNetV3_small_x1_0_ssld", "MobileNetV3_large_x1_0_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"
],
"RegNet": ["RegNetX_4GF"], "RegNet": ["RegNetX_4GF"],
"Res2Net": [ "Res2Net": [
"Res2Net50_14w_8s", "Res2Net50_26w_4s", "Res2Net50_vd_26w_4s", "Res2Net50_14w_8s", "Res2Net50_26w_4s", "Res2Net50_vd_26w_4s",
...@@ -134,6 +139,10 @@ MODEL_SERIES = { ...@@ -134,6 +139,10 @@ MODEL_SERIES = {
"SwinTransformer_small_patch4_window7_224", "SwinTransformer_small_patch4_window7_224",
"SwinTransformer_tiny_patch4_window7_224" "SwinTransformer_tiny_patch4_window7_224"
], ],
"Twins": [
"pcpvt_small", "pcpvt_base", "pcpvt_large", "alt_gvt_small",
"alt_gvt_base", "alt_gvt_large"
],
"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",
...@@ -399,6 +408,7 @@ class PaddleClas(object): ...@@ -399,6 +408,7 @@ class PaddleClas(object):
"""PaddleClas. """PaddleClas.
""" """
init_logger(name='root')
print_info() print_info()
def __init__(self, def __init__(self,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册