- 博客(7)
- 收藏
- 关注
原创 Ubuntu19.04+CUDA10.0+cuDNN7.5.0+anaconda+tensorflow—gpu安装教程 亲测可用
Ubuntu19.04+CUDA+cuDNN+anaconda+tensorflow—gpu安装教程配置:ubuntu19.04显卡:GEFORCE 840M啊啊啊 好垃圾的笔记本。。。没想到还能支持GPU加速 同学们是不是难以置信了Ubuntu19.04安装网络上有很多文章,首先博主是根据B站UP主上的视频一步一步走下来的,有一个视频是讲双系统的,楼主垃圾笔记本一台装了ubuntu和...
2019-05-04 15:17:03 5505 2
原创 将图片按txt格式进行存储,对txt文件批量读取
为什么将图片存储为txt格式在将图片数据输入卷积神经网络时,如果通过np.load进行输入,常常会出现内存爆炸的问题,当然啊,土豪除外。所以讲图片按照txt文件读取,是一种非常好的读取方式,当然啊,如今ternsorflow出的API也相当不错。技多不压身嘛!将图片存储为txt文件代码'''本代码以UC数据集为例子,UC数据集共包含21类,每一类图片包含100张图片其中class_lab...
2019-07-07 22:13:06 1473
原创 对训练数据集进行数据增强代码
对训练数据集进行数据增强代码原代码如下所示:###本代码共采用了四种数据增强,如采用其他数据增强方式,可以参考本代码,随意替换。imageDir 为原数据集的存放位置saveDir 为数据增强后数据的存放位置###def flip(root_path,img_name): #翻转图像 img = Image.open(os.path.join(root_path, ...
2019-07-07 21:53:01 25970 45
原创 对图片数据集按百分比划分为训练数据集和测试集以UC数据集为例
对图片数据集按百分比划分为训练数据集和测试集## 源代码如下## 源代码如下本代码 以UC数据集为例,数据集的文件名称格式为label+编号.tifUC数据集共有21类图片,每一类图片有100张图片,若按照特定百分比去区分数据集,若取80:20 则在程序中标注出更改import numpy as npimport osimport shutilclass_label_dict = {...
2019-07-07 21:42:13 3574 1
原创 解决Failedtoconvert object of type class list
查阅资料发现需要将None 替换成-1output = tf.reshape(input,input.get_shape().as_list(), name="flatten")更改为以下代码shape = input.get_shape().as_list()[1] * input.get_shape().as_list()[2] * input.get_shape().as_list(...
2019-05-27 23:32:05 2014
原创 Ubuntu16.04LST+显卡驱动+tensorflow-GPU标准工作站安装,亲测完美!
Ubuntu16.04LST+tensorflow-GPU标准工作站安装工作站环境说明显卡驱动安装1.禁用nouveau2.安装驱动配置electron-ssr 软件配置本地端口工作站环境说明系统:ubuntu16.04LST显卡:TITAN XPps:工作站内核最好不要更新,工作站稳定第一显卡驱动安装我的第一篇博客介绍了CUDA 以及CUDNN 以及显卡各个版本号之间的关系。有兴趣的...
2019-05-26 10:22:42 784
转载 SCA-CNN tensorflow 源代码
SCA-CNN tensorflow 源代码""" Attention Model: WARNING: Use BatchNorm layer otherwise no accuracy gain. Lower layer with SpatialAttention, high layer with ChannelWiseAttention. In Visual1...
2019-05-16 15:28:10 488 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人