paddlex 2.1.0__py3-none-any.whl → 3.0.0b2__py3-none-any.whl

PaddleClas/ppcls/arch/__init__.py

@@ -1,127 +0,0 @@
-#copyright (c) 2021 PaddlePaddle Authors. All Rights Reserve.
-#
-#Licensed under the Apache License, Version 2.0 (the "License");
-#you may not use this file except in compliance with the License.
-#You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-#Unless required by applicable law or agreed to in writing, software
-#distributed under the License is distributed on an "AS IS" BASIS,
-#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#See the License for the specific language governing permissions and
-#limitations under the License.
-
-import copy
-import importlib
-
-import paddle.nn as nn
-from paddle.jit import to_static
-from paddle.static import InputSpec
-
-from . import backbone, gears
-from .backbone import *
-from .gears import build_gear
-from .utils import *
-from ppcls.arch.backbone.base.theseus_layer import TheseusLayer
-from ppcls.utils import logger
-from ppcls.utils.save_load import load_dygraph_pretrain
-
-__all__ = ["build_model", "RecModel", "DistillationModel"]
-
-
-def build_model(config):
-    config = copy.deepcopy(config)
-    model_type = config.pop("name")
-    mod = importlib.import_module(__name__)
-    arch = getattr(mod, model_type)(**config)
-    return arch
-
-
-def apply_to_static(config, model):
-    support_to_static = config['Global'].get('to_static', False)
-
-    if support_to_static:
-        specs = None
-        if 'image_shape' in config['Global']:
-            specs = [InputSpec([None] + config['Global']['image_shape'])]
-        model = to_static(model, input_spec=specs)
-        logger.info("Successfully to apply @to_static with specs: {}".format(
-            specs))
-    return model
-
-
-class RecModel(nn.Layer):
-    def __init__(self, **config):
-        super().__init__()
-        backbone_config = config["Backbone"]
-        backbone_name = backbone_config.pop("name")
-        self.backbone = eval(backbone_name)(**backbone_config)
-        if "BackboneStopLayer" in config:
-            backbone_stop_layer = config["BackboneStopLayer"]["name"]
-            self.backbone.stop_after(backbone_stop_layer)
-
-        if "Neck" in config:
-            self.neck = build_gear(config["Neck"])
-        else:
-            self.neck = None
-
-        if "Head" in config:
-            self.head = build_gear(config["Head"])
-        else:
-            self.head = None
-
-    def forward(self, x, label=None):
-        x = self.backbone(x)
-        if self.neck is not None:
-            x = self.neck(x)
-        if self.head is not None:
-            y = self.head(x, label)
-        else:
-            y = None
-        return {"features": x, "logits": y}
-
-
-class DistillationModel(nn.Layer):
-    def __init__(self,
-                 models=None,
-                 pretrained_list=None,
-                 freeze_params_list=None,
-                 **kargs):
-        super().__init__()
-        assert isinstance(models, list)
-        self.model_list = []
-        self.model_name_list = []
-        if pretrained_list is not None:
-            assert len(pretrained_list) == len(models)
-
-        if freeze_params_list is None:
-            freeze_params_list = [False] * len(models)
-        assert len(freeze_params_list) == len(models)
-        for idx, model_config in enumerate(models):
-            assert len(model_config) == 1
-            key = list(model_config.keys())[0]
-            model_config = model_config[key]
-            model_name = model_config.pop("name")
-            model = eval(model_name)(**model_config)
-
-            if freeze_params_list[idx]:
-                for param in model.parameters():
-                    param.trainable = False
-            self.model_list.append(self.add_sublayer(key, model))
-            self.model_name_list.append(key)
-
-        if pretrained_list is not None:
-            for idx, pretrained in enumerate(pretrained_list):
-                if pretrained is not None:
-                    load_dygraph_pretrain(
-                        self.model_name_list[idx], path=pretrained)
-
-    def forward(self, x, label=None):
-        result_dict = dict()
-        for idx, model_name in enumerate(self.model_name_list):
-            if label is None:
-                result_dict[model_name] = self.model_list[idx](x)
-            else:
-                result_dict[model_name] = self.model_list[idx](x, label)
-        return result_dict

PaddleClas/ppcls/arch/backbone/__init__.py

@@ -1,80 +0,0 @@
-# copyright (c) 2021 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import sys
-import inspect
-
-from ppcls.arch.backbone.legendary_models.mobilenet_v1 import MobileNetV1_x0_25, MobileNetV1_x0_5, MobileNetV1_x0_75, MobileNetV1
-from ppcls.arch.backbone.legendary_models.mobilenet_v3 import MobileNetV3_small_x0_35, MobileNetV3_small_x0_5, MobileNetV3_small_x0_75, MobileNetV3_small_x1_0, MobileNetV3_small_x1_25, MobileNetV3_large_x0_35, MobileNetV3_large_x0_5, MobileNetV3_large_x0_75, MobileNetV3_large_x1_0, MobileNetV3_large_x1_25
-from ppcls.arch.backbone.legendary_models.resnet import ResNet18, ResNet18_vd, ResNet34, ResNet34_vd, ResNet50, ResNet50_vd, ResNet101, ResNet101_vd, ResNet152, ResNet152_vd, ResNet200_vd
-from ppcls.arch.backbone.legendary_models.vgg import VGG11, VGG13, VGG16, VGG19
-from ppcls.arch.backbone.legendary_models.inception_v3 import InceptionV3
-from ppcls.arch.backbone.legendary_models.hrnet import HRNet_W18_C, HRNet_W30_C, HRNet_W32_C, HRNet_W40_C, HRNet_W44_C, HRNet_W48_C, HRNet_W60_C, HRNet_W64_C, SE_HRNet_W64_C
-from ppcls.arch.backbone.legendary_models.pp_lcnet import PPLCNet_x0_25, PPLCNet_x0_35, PPLCNet_x0_5, PPLCNet_x0_75, PPLCNet_x1_0, PPLCNet_x1_5, PPLCNet_x2_0, PPLCNet_x2_5
-from ppcls.arch.backbone.legendary_models.esnet import ESNet_x0_25, ESNet_x0_5, ESNet_x0_75, ESNet_x1_0
-
-from ppcls.arch.backbone.model_zoo.resnet_vc import ResNet50_vc
-from ppcls.arch.backbone.model_zoo.resnext import ResNeXt50_32x4d, ResNeXt50_64x4d, ResNeXt101_32x4d, ResNeXt101_64x4d, ResNeXt152_32x4d, ResNeXt152_64x4d
-from ppcls.arch.backbone.model_zoo.resnext_vd import ResNeXt50_vd_32x4d, ResNeXt50_vd_64x4d, ResNeXt101_vd_32x4d, ResNeXt101_vd_64x4d, ResNeXt152_vd_32x4d, ResNeXt152_vd_64x4d
-from ppcls.arch.backbone.model_zoo.res2net import Res2Net50_26w_4s, Res2Net50_14w_8s
-from ppcls.arch.backbone.model_zoo.res2net_vd import Res2Net50_vd_26w_4s, Res2Net101_vd_26w_4s, Res2Net200_vd_26w_4s
-from ppcls.arch.backbone.model_zoo.se_resnet_vd import SE_ResNet18_vd, SE_ResNet34_vd, SE_ResNet50_vd
-from ppcls.arch.backbone.model_zoo.se_resnext_vd import SE_ResNeXt50_vd_32x4d, SE_ResNeXt50_vd_32x4d, SENet154_vd
-from ppcls.arch.backbone.model_zoo.se_resnext import SE_ResNeXt50_32x4d, SE_ResNeXt101_32x4d, SE_ResNeXt152_64x4d
-from ppcls.arch.backbone.model_zoo.dpn import DPN68, DPN92, DPN98, DPN107, DPN131
-from ppcls.arch.backbone.model_zoo.densenet import DenseNet121, DenseNet161, DenseNet169, DenseNet201, DenseNet264
-from ppcls.arch.backbone.model_zoo.efficientnet import EfficientNetB0, EfficientNetB1, EfficientNetB2, EfficientNetB3, EfficientNetB4, EfficientNetB5, EfficientNetB6, EfficientNetB7, EfficientNetB0_small
-from ppcls.arch.backbone.model_zoo.resnest import ResNeSt50_fast_1s1x64d, ResNeSt50, ResNeSt101
-from ppcls.arch.backbone.model_zoo.googlenet import GoogLeNet
-from ppcls.arch.backbone.model_zoo.mobilenet_v2 import MobileNetV2_x0_25, MobileNetV2_x0_5, MobileNetV2_x0_75, MobileNetV2, MobileNetV2_x1_5, MobileNetV2_x2_0
-from ppcls.arch.backbone.model_zoo.shufflenet_v2 import ShuffleNetV2_x0_25, ShuffleNetV2_x0_33, ShuffleNetV2_x0_5, ShuffleNetV2_x1_0, ShuffleNetV2_x1_5, ShuffleNetV2_x2_0, ShuffleNetV2_swish
-from ppcls.arch.backbone.model_zoo.ghostnet import GhostNet_x0_5, GhostNet_x1_0, GhostNet_x1_3
-from ppcls.arch.backbone.model_zoo.alexnet import AlexNet
-from ppcls.arch.backbone.model_zoo.inception_v4 import InceptionV4
-from ppcls.arch.backbone.model_zoo.xception import Xception41, Xception65, Xception71
-from ppcls.arch.backbone.model_zoo.xception_deeplab import Xception41_deeplab, Xception65_deeplab
-from ppcls.arch.backbone.model_zoo.resnext101_wsl import ResNeXt101_32x8d_wsl, ResNeXt101_32x16d_wsl, ResNeXt101_32x32d_wsl, ResNeXt101_32x48d_wsl
-from ppcls.arch.backbone.model_zoo.squeezenet import SqueezeNet1_0, SqueezeNet1_1
-from ppcls.arch.backbone.model_zoo.darknet import DarkNet53
-from ppcls.arch.backbone.model_zoo.regnet import RegNetX_200MF, RegNetX_4GF, RegNetX_32GF, RegNetY_200MF, RegNetY_4GF, RegNetY_32GF
-from ppcls.arch.backbone.model_zoo.vision_transformer import ViT_small_patch16_224, ViT_base_patch16_224, ViT_base_patch16_384, ViT_base_patch32_384, ViT_large_patch16_224, ViT_large_patch16_384, ViT_large_patch32_384, ViT_huge_patch16_224, ViT_huge_patch32_384
-from ppcls.arch.backbone.model_zoo.distilled_vision_transformer import DeiT_tiny_patch16_224, DeiT_small_patch16_224, DeiT_base_patch16_224, DeiT_tiny_distilled_patch16_224, DeiT_small_distilled_patch16_224, DeiT_base_distilled_patch16_224, DeiT_base_patch16_384, DeiT_base_distilled_patch16_384
-from ppcls.arch.backbone.model_zoo.swin_transformer import SwinTransformer_tiny_patch4_window7_224, SwinTransformer_small_patch4_window7_224, SwinTransformer_base_patch4_window7_224, SwinTransformer_base_patch4_window12_384, SwinTransformer_large_patch4_window7_224, SwinTransformer_large_patch4_window12_384
-from ppcls.arch.backbone.model_zoo.mixnet import MixNet_S, MixNet_M, MixNet_L
-from ppcls.arch.backbone.model_zoo.rexnet import ReXNet_1_0, ReXNet_1_3, ReXNet_1_5, ReXNet_2_0, ReXNet_3_0
-from ppcls.arch.backbone.model_zoo.gvt import pcpvt_small, pcpvt_base, pcpvt_large, alt_gvt_small, alt_gvt_base, alt_gvt_large
-from ppcls.arch.backbone.model_zoo.levit import LeViT_128S, LeViT_128, LeViT_192, LeViT_256, LeViT_384
-from ppcls.arch.backbone.model_zoo.dla import DLA34, DLA46_c, DLA46x_c, DLA60, DLA60x, DLA60x_c, DLA102, DLA102x, DLA102x2, DLA169
-from ppcls.arch.backbone.model_zoo.rednet import RedNet26, RedNet38, RedNet50, RedNet101, RedNet152
-from ppcls.arch.backbone.model_zoo.tnt import TNT_small
-from ppcls.arch.backbone.model_zoo.hardnet import HarDNet68, HarDNet85, HarDNet39_ds, HarDNet68_ds
-from ppcls.arch.backbone.model_zoo.cspnet import CSPDarkNet53
-from ppcls.arch.backbone.variant_models.resnet_variant import ResNet50_last_stage_stride1
-from ppcls.arch.backbone.variant_models.vgg_variant import VGG19Sigmoid
-from ppcls.arch.backbone.variant_models.pp_lcnet_variant import PPLCNet_x2_5_Tanh
-
-
-def get_apis():
-    current_func = sys._getframe().f_code.co_name
-    current_module = sys.modules[__name__]
-    api = []
-    for _, obj in inspect.getmembers(current_module,
-                                     inspect.isclass) + inspect.getmembers(
-                                         current_module, inspect.isfunction):
-        api.append(obj.__name__)
-    api.remove(current_func)
-    return api
-
-
-__all__ = get_apis()

PaddleClas/ppcls/arch/backbone/base/theseus_layer.py

@@ -1,126 +0,0 @@
-from abc import ABC
-from paddle import nn
-import re
-
-
-class Identity(nn.Layer):
-    def __init__(self):
-        super(Identity, self).__init__()
-
-    def forward(self, inputs):
-        return inputs
-
-
-class TheseusLayer(nn.Layer):
-    def __init__(self, *args, **kwargs):
-        super(TheseusLayer, self).__init__()
-        self.res_dict = {}
-        self.res_name = self.full_name()
-
-    # stop doesn't work when stop layer has a parallel branch.
-    def stop_after(self, stop_layer_name: str):
-        after_stop = False
-        for layer_i in self._sub_layers:
-            if after_stop:
-                self._sub_layers[layer_i] = Identity()
-                continue
-            layer_name = self._sub_layers[layer_i].full_name()
-            if layer_name == stop_layer_name:
-                after_stop = True
-                continue
-            if isinstance(self._sub_layers[layer_i], TheseusLayer):
-                after_stop = self._sub_layers[layer_i].stop_after(
-                    stop_layer_name)
-        return after_stop
-
-    def update_res(self, return_patterns):
-        for return_pattern in return_patterns:
-            pattern_list = return_pattern.split(".")
-            if not pattern_list:
-                continue
-            sub_layer_parent = self
-            while len(pattern_list) > 1:
-                if '[' in pattern_list[0]:
-                    sub_layer_name = pattern_list[0].split('[')[0]
-                    sub_layer_index = pattern_list[0].split('[')[1].split(']')[0]
-                    sub_layer_parent = getattr(sub_layer_parent, sub_layer_name)[sub_layer_index]
-                else:
-                    sub_layer_parent = getattr(sub_layer_parent, pattern_list[0],
-                                               None)
-                    if sub_layer_parent is None:
-                        break
-                if isinstance(sub_layer_parent, WrapLayer):
-                    sub_layer_parent = sub_layer_parent.sub_layer
-                pattern_list = pattern_list[1:]
-            if sub_layer_parent is None:
-                continue
-            if '[' in pattern_list[0]:
-                sub_layer_name = pattern_list[0].split('[')[0]
-                sub_layer_index = pattern_list[0].split('[')[1].split(']')[0]
-                sub_layer = getattr(sub_layer_parent, sub_layer_name)[sub_layer_index]
-                if not isinstance(sub_layer, TheseusLayer):
-                    sub_layer = wrap_theseus(sub_layer)
-                getattr(sub_layer_parent, sub_layer_name)[sub_layer_index] = sub_layer
-            else:
-                sub_layer = getattr(sub_layer_parent, pattern_list[0])
-                if not isinstance(sub_layer, TheseusLayer):
-                    sub_layer = wrap_theseus(sub_layer)
-                setattr(sub_layer_parent, pattern_list[0], sub_layer)
-
-            sub_layer.res_dict = self.res_dict
-            sub_layer.res_name = return_pattern
-            sub_layer.register_forward_post_hook(sub_layer._save_sub_res_hook)
-
-    def _save_sub_res_hook(self, layer, input, output):
-        self.res_dict[self.res_name] = output
-
-    def _return_dict_hook(self, layer, input, output):
-        res_dict = {"output": output}
-        for res_key in list(self.res_dict):
-            res_dict[res_key] = self.res_dict.pop(res_key)
-        return res_dict
-
-    def replace_sub(self, layer_name_pattern, replace_function,
-                    recursive=True):
-        for layer_i in self._sub_layers:
-            layer_name = self._sub_layers[layer_i].full_name()
-            if re.match(layer_name_pattern, layer_name):
-                self._sub_layers[layer_i] = replace_function(self._sub_layers[
-                    layer_i])
-            if recursive:
-                if isinstance(self._sub_layers[layer_i], TheseusLayer):
-                    self._sub_layers[layer_i].replace_sub(
-                        layer_name_pattern, replace_function, recursive)
-                elif isinstance(self._sub_layers[layer_i],
-                                (nn.Sequential, nn.LayerList)):
-                    for layer_j in self._sub_layers[layer_i]._sub_layers:
-                        self._sub_layers[layer_i]._sub_layers[
-                            layer_j].replace_sub(layer_name_pattern,
-                                                 replace_function, recursive)
-
-    '''
-    example of replace function:
-    def replace_conv(origin_conv: nn.Conv2D):
-        new_conv = nn.Conv2D(
-            in_channels=origin_conv._in_channels,
-            out_channels=origin_conv._out_channels,
-            kernel_size=origin_conv._kernel_size,
-            stride=2
-        )
-        return new_conv
-
-        '''
-
-
-class WrapLayer(TheseusLayer):
-    def __init__(self, sub_layer):
-        super(WrapLayer, self).__init__()
-        self.sub_layer = sub_layer
-
-    def forward(self, *inputs, **kwargs):
-        return self.sub_layer(*inputs, **kwargs)
-
-
-def wrap_theseus(sub_layer):
-    wrapped_layer = WrapLayer(sub_layer)
-    return wrapped_layer

PaddleClas/ppcls/arch/backbone/legendary_models/__init__.py

@@ -1,6 +0,0 @@
-from .resnet import ResNet18, ResNet34, ResNet50, ResNet101, ResNet152, ResNet18_vd, ResNet34_vd, ResNet50_vd, ResNet101_vd, ResNet152_vd
-from .hrnet import HRNet_W18_C, HRNet_W30_C, HRNet_W32_C, HRNet_W40_C, HRNet_W44_C, HRNet_W48_C, HRNet_W64_C
-from .mobilenet_v1 import MobileNetV1_x0_25, MobileNetV1_x0_5, MobileNetV1_x0_75, MobileNetV1
-from .mobilenet_v3 import MobileNetV3_small_x0_35, MobileNetV3_small_x0_5, MobileNetV3_small_x0_75, MobileNetV3_small_x1_0, MobileNetV3_small_x1_25, MobileNetV3_large_x0_35, MobileNetV3_large_x0_5, MobileNetV3_large_x0_75, MobileNetV3_large_x1_0, MobileNetV3_large_x1_25
-from .inception_v3 import InceptionV3
-from .vgg import VGG11, VGG13, VGG16, VGG19