mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Revert "backbone_zoo -> backbone" 

This reverts commit e4a3e1bb12.
pull/2701/head
Tingquan Gao 2023-03-14 16:16:40 +08:00
parent f42719afbb
commit 502aead98b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ppcls/arch/__init__.py
View File

@ -19,7 +19,7 @@ import paddle.nn as nn
from paddle.jit import to_static
from paddle.static import InputSpec
from . import backbone
from . import backbone as backbone_zoo
from .gears import build_gear
from .utils import *
from .backbone.base.theseus_layer import TheseusLayer
@ -36,7 +36,7 @@ def build_model(config, mode="train"):
model_type = arch_config.pop("name")
use_sync_bn = arch_config.pop("use_sync_bn", False)
if hasattr(backbone, model_type):
if hasattr(backbone_zoo, model_type):
model = ClassModel(model_type, **arch_config)
else:
model = getattr(sys.modules[__name__], model_type)("ClassModel",
@ -83,7 +83,7 @@ class ClassModel(TheseusLayer):
else:
backbone_name = model_type
backbone_config = config
self.backbone = getattr(backbone, backbone_name)(**backbone_config)
self.backbone = getattr(backbone_zoo, backbone_name)(**backbone_config)
def forward(self, batch):
x, label = batch[0], batch[1]
@ -95,7 +95,7 @@ class RecModel(TheseusLayer):
super().__init__()
backbone_config = config["Backbone"]
backbone_name = backbone_config.pop("name")
self.backbone = getattr(backbone, backbone_name)(**backbone_config)
self.backbone = getattr(backbone_zoo, backbone_name)(**backbone_config)
self.head_feature_from = config.get('head_feature_from', 'neck')
if "BackboneStopLayer" in config: