提交 d05d9d8d 编写于 作者: W weishengyu

update functions

上级 be220a5f
import paddle import paddle
from ppcls.arch import build_model from ppcls.arch import build_model
from deploy.utils.config import parse_config, parse_args from ppcls.utils.config import parse_config, parse_args
def load_feature_extractor(configs):
pass
def build_gallery_feature(feature_extractor): def build_gallery_feature(feature_extractor):
...@@ -18,7 +14,7 @@ def save_fuse_model(fuse_model): ...@@ -18,7 +14,7 @@ def save_fuse_model(fuse_model):
class FuseModel(paddle.nn.Layer): class FuseModel(paddle.nn.Layer):
def __init__(self, configs): def __init__(self, configs):
super().__init__() super().__init__()
self.feature_extractor = load_feature_extractor(configs) self.feature_extractor = build_model(configs)
self.gallery_layer = build_gallery_feature(self.feature_extractor) self.gallery_layer = build_gallery_feature(self.feature_extractor)
def forward(self, x): def forward(self, x):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册