python
python
青春小尾巴-
这个作者很懒,什么都没留下…
展开
-
tensorflow的ckpt文件转为npy文件
我们迁移模型参数时,需要将ckpt文件转为npy文件,以VGG19为例:"""将ckpt文件转化为npy文件"""import numpy as npimport tensorflow as tffrom tensorflow.python import pywrap_tensorflowdef ckpt2npy(): checkpoint_path = 'model/model.ckpt-4999' reader = pywrap_tensorflow.NewCheck原创 2021-01-18 16:59:36 · 538 阅读 · 0 评论 -
python数组进行降维
在深度学习训练过程中,我们有时候想要输出图片看看图片长什么样,但是训练时的图片格式一般都会多出一个批次的维度,如[1,400,400,3],这时我们需要把第1个维度去掉才能展示图片:如果是array的格式,可以直接img = img.squeeze...原创 2021-04-07 14:37:42 · 2577 阅读 · 0 评论 -
pycharm 调试功能实现跳到循环的某一步
可以使用条件断点,在断点上右键可以设置运行停止的条件,这样代码就会一直运行至你设置的条件处,再进入debug模式原创 2021-04-07 14:51:04 · 1565 阅读 · 1 评论