keras 框架与 django 框架一起使用 反复调用 model 模型 出错 - V2EX
V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
4thmagi
V2EX    TensorFlow

keras 框架与 django 框架一起使用 反复调用 model 模型 出错

  •  
  •   4thmagi 2018-06-12 23:22:32 +08:00 4982 次点击
    这是一个创建于 2740 天前的主题,其中的信息可能已经有所发展或是发生改变。
    我写了一个网页,是基于 django 的。该网页通过 session 实现了用户登录状态的判断。用户在登录情况下可以发布信息,其中可以包含图片。然后可以对发布的图片进行数字的识别。
    网页后端我使用的是 django 框架,上传一张图片传入基于 tensorflow 的 keras 模型进行预测,重复预测时,报告错误:ValueError: Tensor Tensor("Placeholder:0", shape=(3, 3, 1, 32), dtype=float32)。查了一下,原因大概是第二次预测时,model 底层 tensorflow 的 session 中还有数据。
    在网上找到的解决方法是调用模型前面加一句 keras.backend.clear_session()。但是这样的话就把我的登录状态也清空了,网页直接崩溃了。提示 python 已停止工作。
    请问有什么办法可以只清空 session 中的 TensorFlow 的相关数据?
    5 条回复    2018-06-13 16:25:15 +08:00
    4thmagi
        1
    4thmagi  
    OP
       2018-06-12 23:26:51 +08:00
    keras 除了 keras.backend.clear_session()之外有方法选择性清除 session 吗
    ipwx
        2
    ipwx  
       2018-06-13 00:13:31 +08:00 via iPhone   1
    拿到 Tensorflow 的输出,直接用 session.run 。这部分做成单独的 API 供调用。可以参考 tensorflow-serve。别用 keras 封装的什么 predict,细节都被掩盖了,不好改
    takato
        3
    takato  
       2018-06-13 01:18:22 +08:00
    Keras 属于上手容易。。。但是你真的用起来。。实在是不好用。。
    4thmagi
        4
    4thmagi  
    OP
       2018-06-13 10:05:41 +08:00
    @ipwx 我在每次进行上传图片进行预测的时候都会 load_model 一次,我怀疑是在 load_model 时因为上次识别已 load 过了,所以再次识别图片时,在 load 这里出错了。有没有办法在启动 django 的时候直接将训练好的模型写入 session,通过这样来避免每次识别时写入导致的出错。
    ipwx
        5
    ipwx  
       2018-06-13 16:25:15 +08:00 via iPhone   1
    @4thmagi 还是建议参考 tensorflow-serving,好歹官方出品。要我自己手写,我会写个 while True socket,单独运行,做微服务,然后 django 负责转接这个微服务。当然也可以考虑微服务上多线程多进程。不管则么样我不会直接用 django 调模型
    关于     帮助文档     自助推广系统     博客     API     FAQ     Solana     2520 人在线   最高记录 6679       Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 24ms UTC 04:24 PVG 12:24 LAX 20:24 JFK 23:24
    Do have faith in what you're doing.
    ubao msn snddm index pchome yahoo rakuten mypaper meadowduck bidyahoo youbao zxmzxm asda bnvcg cvbfg dfscv mmhjk xxddc yybgb zznbn ccubao uaitu acv GXCV ET GDG YH FG BCVB FJFH CBRE CBC GDG ET54 WRWR RWER WREW WRWER RWER SDG EW SF DSFSF fbbs ubao fhd dfg ewr dg df ewwr ewwr et ruyut utut dfg fgd gdfgt etg dfgt dfgd ert4 gd fgg wr 235 wer3 we vsdf sdf gdf ert xcv sdf rwer hfd dfg cvb rwf afb dfh jgh bmn lgh rty gfds cxv xcv xcs vdas fdf fgd cv sdf tert sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf shasha9178 shasha9178 shasha9178 shasha9178 shasha9178 liflif2 liflif2 liflif2 liflif2 liflif2 liblib3 liblib3 liblib3 liblib3 liblib3 zhazha444 zhazha444 zhazha444 zhazha444 zhazha444 dende5 dende denden denden2 denden21 fenfen9 fenf619 fen619 fenfe9 fe619 sdf sdf sdf sdf sdf zhazh90 zhazh0 zhaa50 zha90 zh590 zho zhoz zhozh zhozho zhozho2 lislis lls95 lili95 lils5 liss9 sdf0ty987 sdft876 sdft9876 sdf09876 sd0t9876 sdf0ty98 sdf0976 sdf0ty986 sdf0ty96 sdf0t76 sdf0876 df0ty98 sf0t876 sd0ty76 sdy76 sdf76 sdf0t76 sdf0ty9 sdf0ty98 sdf0ty987 sdf0ty98 sdf6676 sdf876 sd876 sd876 sdf6 sdf6 sdf9876 sdf0t sdf06 sdf0ty9776 sdf0ty9776 sdf0ty76 sdf8876 sdf0t sd6 sdf06 s688876 sd688 sdf86