![](https://img-blog.csdnimg.cn/20201014180756754.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
计算机视觉
chyelang12
个人博客:https://blog.chyelang.ml
展开
-
基于卷积神经网络进行图像分类
利用TensorFlow(TF),本项目实现了一个基于full convolution stack、inception v2 module等模块的图片分类网络,纵向共包含10层包含参数的层,采用自己搭建的卷积层。对于dset1与dset2两个数据集采用同样的网络结构,分别进行训练。最终在dset1验证集(约含900张图,下同)上的Top1分类准确率约为0.52,在dset2验证集上的Top1准确率约为0.63。在K80显卡的单核上,该模型的训练速度约为205张图每秒,最终所得模型的checkpoint约为4原创 2018-12-18 23:11:51 · 20507 阅读 · 6 评论 -
TensorFlow 1.13.1 CUDNN_STATUS_ALLOC_FAILED 错误解决办法
为使用faceswap玩一下换脸,在win10+rtx2070平台上搭建TensorFlow-gpu 1.13.1环境。安装完毕后import tensorflow没有问题,但在运行时总是报Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED错误,上网多方查证后认为cuda、cudnn、TensorFlow的版本搭配没有问题,但就是反复重...原创 2019-04-18 21:45:04 · 1603 阅读 · 0 评论