随笔分类 -  Tensorflow

摘要:tensorflow给出的压缩 :https://www.tensorflow.org/model_optimization/guide/pruning/pruning_with_keras 阅读全文
posted @ 2019-11-22 21:25 闪存第一菜鸡 阅读(176) 评论(0) 推荐(0)
摘要:tf.add_to_collection() tf.get_collection() tf.add_n() tf.add_to_collection:把变量放入一个集合,把很多变量变成一个列表 tf.get_collection:从一个结合中取出全部变量,是一个列表 tf.add_n:把一个列表的东 阅读全文
posted @ 2019-11-07 18:59 闪存第一菜鸡 阅读(185) 评论(0) 推荐(0)
摘要:出现这个的原因是gpu显存不够导致的,一般是我们设置了程序根据需求增长导致的 这里,我们设置 allow_growth=False 就可以控制显存使用的增长,从而控制显存的使用,而不会程序运行一半报错。 这种情况是建立在我们的显存比较小的情况下使用的策略,如果说我们显存够大则不用担心这种问题。 阅读全文
posted @ 2019-11-07 18:50 闪存第一菜鸡 阅读(1152) 评论(0) 推荐(0)
摘要:1.tensorflow 读取 图片并灰度化 这里其实有个要注意的,不同于PIL.image 和 cv 的图片读取,这里会带上通道! 阅读全文
posted @ 2019-10-15 11:03 闪存第一菜鸡 阅读(459) 评论(0) 推荐(0)