提交 a1ba7603 编写于 作者: W weishengyu

init fuse gallery

上级 c4cf0348
import paddle
from deploy.utils.config import parse_config, parse_args
def load_feature_extractor(configs):
pass
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)
fuse_model = FuseModel(configs)
save_fuse_model(fuse_model)
if __name__ == '__main__':
main()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册