43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
import paddle
|
|
from ppcls.arch import build_model
|
|
from ppcls.utils.config import parse_config, parse_args
|
|
from ppcls.utils.save_load import load_dygraph_pretrain
|
|
from ppcls.utils.logger import init_logger
|
|
|
|
|
|
def load_feature_extractor(configs):
|
|
arch = build_model(configs["Arch"])
|
|
load_dygraph_pretrain(arch, configs["Global"]["pretrained_model"])
|
|
|
|
|
|
def build_gallery_feature(feature_extractor):
|
|
pass
|
|
|
|
|
|
def save_fuse_model(fuse_model):
|
|
pass
|
|
|
|
|
|
class FuseModel(paddle.nn.Layer):
|
|
def __init__(self, configs):
|
|
super().__init__()
|
|
self.feature_extractor = load_feature_extractor(configs)
|
|
self.gallery_layer = build_gallery_feature(self.feature_extractor)
|
|
|
|
def forward(self, x):
|
|
x = self.feature_model(x)
|
|
x = self.gallery_layer(x)
|
|
return x
|
|
|
|
|
|
def main():
|
|
args = parse_args()
|
|
configs = parse_config(args.config)
|
|
init_logger(name='gallery2fc')
|
|
fuse_model = FuseModel(configs)
|
|
save_fuse_model(fuse_model)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|