- 博客(4)
- 收藏
- 关注
RSS订阅原创 mnist数据处理
import torch as timport torch.nn as nnimport numpy as npimport os.pathimport structtrain_img_dir_ubyte = r'E:\code\LnNet\train-images.idx3-ubyte'train_label_dir_ubyte = r'E:\code\LnNet\train-labels.idx1-ubyte'train_size = os.path.getsize(train_im.
2022-01-04 10:13:41
1922
原创 ubantu18.04下基于cuda11.0的MaskRCNN环境搭建
介绍一下我的电脑是显卡3080ti，好像只能支持cuda11.0以上。自己摸索最后通过了balloon气球的实验。记录一下。nvcc -V显示是cuda11.1 cudnn是8.0.51.anaconda安装自己在网上百度一下，不再赘述。2.tensorflow-gpu keras安装cuda=11.0 cudnn=8.0.5 支持的tensorflow-gpu是2.4 tensorflow-gpu支持的keras=2.4.3.不过在tensorflow2以后好像keras的一部分功能都已经融入t
2021-11-21 14:45:44
1327
1
原创 随机分配数据集和验证集
找了很多代码 不好用调试没成功 最后逼得小白自己写了一个代码如下 需要自取import osimport randomimport shutil'''数据集划分： 图片在一个文件夹 需要四个文件夹 注意：图片的文件里只能有数据图片 标签在另一个文件夹 训练集在一个文件夹 测试集在一个文件夹 最后生成 train.txt train_xml.txt 文件 test.t
2021-11-01 21:55:21
374
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人