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

update functions

pull/1599/head
weishengyu 2021-12-08 19:48:59 +08:00
parent be220a5f09
commit d05d9d8d65
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ppcls/utils/gallery2fc.py
View File

@ -1,10 +1,6 @@
import paddle
from ppcls.arch import build_model
from deploy.utils.config import parse_config, parse_args
def load_feature_extractor(configs):
pass
from ppcls.utils.config import parse_config, parse_args
def build_gallery_feature(feature_extractor):
@ -18,7 +14,7 @@ def save_fuse_model(fuse_model):
class FuseModel(paddle.nn.Layer):
def __init__(self, configs):
super().__init__()
self.feature_extractor = load_feature_extractor(configs)
self.feature_extractor = build_model(configs)
self.gallery_layer = build_gallery_feature(self.feature_extractor)
def forward(self, x):