
tensorflow
Ricky_Yan
这个作者很懒,什么都没留下…
展开
-
Keras、TensorFlow、PyTorch的python程序中止Aborted (core dumped)
plt导致python程序终止Aborted (core dumped)用Keras、TensorFlow、PyTorch的模型预测时,如果用了matplotlib显示图片,一直循环会导致plt占内存过多:terminate called after throwing an instance of ‘std::bad_alloc’Aborted (core dumped)各种尝试分析问题不在Keras、TensorFlow、PyTorch,而在于matplotlib打开figure过多因此用:原创 2020-08-10 19:11:05 · 2307 阅读 · 0 评论 -
把Python变量记录到TensorFlow的summary中
把Python变量记录到TensorFlow的summary中在用tensorboard可视化时,遇到过一个坑:打算每训练一定步数,将验证集准确率记录下来。一般做法如下:...# 1、计算准确率correct_prediction = tf.equal(tf.argmax(logits, 1), tf.argmax(y, 1))accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32)) # 2、添加scalar,汇总转载 2020-05-27 14:35:04 · 251 阅读 · 0 评论 -
opencv用dnn.readNet加载caffe/torch/darknet/tensorflow的模型和权重
这里写自定义目录标题cv.dnn.readNet()cv.dnn 的其它加载模型的方法有:cv.dnn.readNet()官方文档:https://docs.opencv.org/3.4/d6/d0f/group__dnn.html#ga3b34fe7a29494a6a4295c169a7d32422模型权重文件支持的文件格式:*.caffemodel (Caffe, http://c...原创 2019-12-31 16:21:31 · 14857 阅读 · 4 评论