小小程序天-CSDN博客

原创 python实现将给定路径下全部文件的绝对路径写入到txt文件中

将给定路径下全部文件的路径写入到txt文件中再跑深度学习代码时，读取文件常常以txt文件存储文件路径。该代码实现了将当前路径写入到txt文件中，实现了很好的读取文件。import globimport csvimport ospath = ''file_list = sorted(glob.glob(os.path.join(path, '*')))f = open('data.txt', 'w')csv_writer = csv.writer(f)for i in range(len(

2022-03-11 11:20:30 2224 1

原创地震数据.dat文件转.sgy文件

地震数据.dat文件转.sgy文件最近接触了地震去噪任务，需要使用dat文件和sgy文件，所以常会遇到两者的转换，现将代码板书如下：import numpy as npimport segyioimport osif __name__ == '__main__': source_file = '' # 源文件路径 save_dir = '' # 文件保存目录 trace_n = # 数据总道数 sample_n = # 数据采样点数 sample_i

2021-09-09 17:01:18 2189 8

原创 pytorch实现全连接网络的vae

网络全用全连接层nn.Linear()import torchimport torch.nn as nnimport torchvisionimport torchvision.transforms as transformsimport torch.optim as optimimport osclass VAE(nn.Module): def __init__(self): super(VAE, self).__init__() self.E_f

2020-10-25 16:46:59 986

原创 pytorch实现批量resize图片尺寸

python实现对图片尺寸进行resize的完整代码,包含对一个目录下的所有图片的读取,resize,和重新存储:import torch.utils.data as datafrom glob import globfrom PIL import Imageimport torchvision.transforms as transformsimport argparseimport osimport imageioclass DataSet(data.Dataset): de

2020-09-04 09:51:36 7650 9

原创 pytorch实现训练Cifar10

import torch.nn as nnimport torch.nn.functional as Fimport torchvisionimport torchvision.transforms as transformsimport torch.optim as optimimport torchclass LeNet(nn.Module): # 构造方法里面定义可学...

2020-04-09 09:41:57 1103

原创 pytorch实现LeNet网络结构

import torch.nn as nnimport torch.nn.functional as Fimport torchimport torch.optim as optimclass LeNet(nn.Module): # 构造方法里面定义可学习参数的层 def __init__(self): super(LeNet, self).__init...

2020-04-09 09:39:47 215

原创 python 将文件夹下的所有文件全部复制到目标文件夹下

linux系统下，其实可以直接使用：“cp /源文件夹/* /目的文件夹” 这条命令来复制文件，但是但源文件夹文件数过多时，如文件有几十万甚至更多时，该命令就无效了，会提示参数过多，此时就可以使用python代码来复制文件夹import osimport shutilsrc = "" # 源文件目录det = "" # 目的文件目录for root, _, fnames in os...

2020-01-06 22:18:39 5317

原创推荐一个查看linux gpu的命令个人觉得比nvidia-smi 好用

先用pip install gpustat，安装成功后就可以用 gpustat --watch 查看gpu了主要是觉得字体颜色和信息比nvidia-smi显示的内容要好些

2019-12-24 22:42:51 677

原创 S3DIS 点云数据集的手动修复问题

有人用过S3DIS 这个点云数据集吗？在它的Stanford3dDataset_v1.2_Aligned_Version 这个版本的数据集中，有个Area_5/hallway_6数据需要手动修复一下，有人知道怎么做的吗？谢谢！补充一下我是在跑GitHub上的ASIS的代码时用到的：Associatively Segmenting Instances and Semantics in Point ...

2019-08-29 14:47:07 2490 18

qq_42115159的博客