![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
狮子座的尾巴~
这个作者很懒,什么都没留下…
展开
-
解决tensorflow.python.framework.errors_impl.UnknownError
解决tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.在代码前加入以下代码即可from tensorflow import ConfigPr原创 2021-01-04 16:12:49 · 3348 阅读 · 1 评论 -
深度学习中的图像分割标准DICE系数
深度学习中的图像分割标准DICE系数dice系数的取值范围为0-1,越接近1说明构建的模型越好Dice系数是像素级别的,真实的目标(groundtruth)出现在某片区域A,我们的模型预测结果的目标区域为B,那么Dice系数公式为:python代码为# 定义Dice系数def dice_coef(y_true, y_pred): y_truef = K.flatten(y_true) y_predf = K.flatten(y_pred) And = K.sum(y_tr原创 2020-11-21 20:57:56 · 10462 阅读 · 1 评论 -
跑深度学习时tensorflow-gpu内存不够使用如何解决
跑深度学习时tensorflow-gpu不够使用如何解决今天跑一个深度学习代码时遇到一个问题:Hint: If you want to see a list of allocated tensors when OOM happen百度了之后发现是gpu内存不够使用两个解决方案1.制定第二块GPU可用import osos.environ["CUDA_VISIBLE_DEVICES"] = "2"2.调整batch_size的大小400张训练集,我的batch_size原本为8,跑50次出原创 2020-11-21 19:38:26 · 5905 阅读 · 4 评论