python
Virogous
抖动产生、测试
展开
-
PYTHON 随机分类自建训练集和测试集,以及对应的标签label
描述本文想解决的是在神经网络训练中,需要随机划分训练集和测试集,使得训练集可以更好的覆盖测试集的均值和方差范围。其中,(1)数据是存放在csv文件中,因为网络的输入是二维的point cloud(2)标签同样是放在一个csv文件中,csv文件共四列, 第一列是csv数据文件的名字,第二,第三和第四列是需要预测的数值。导入包在这篇文章中,我们需要用到os库 创建文件夹,定位标签文件的pathcsv库 用来处理csv文件csv文件批量处理可以参考这一篇博客random 用来产生随机数shu原创 2020-12-24 11:44:20 · 4063 阅读 · 0 评论 -
Pytorch-IndexError: too many indices for tensor of dimension 0
在采用pytorch进行CNN实现了,遇到了如下的问题:IndexError: too many indices for tensor of dimension 0具体代码如下:JitterTest_Dataset = JitterDataset( csv_file= './jitter_labels/new_jitter_test.CSV', root_dir= './Test_jitter_Square/', transform= transforms.Compose([原创 2020-11-30 15:53:11 · 5320 阅读 · 3 评论 -
Pytorch-RuntimeError: size mismatch m1: [a x b], m2: [c x d]
Pytorch-RuntimeError: size mismatch m1: [a x b], m2: [c x d]在进行深度学习的代码中,在采用线性层进行转化的时候可能出现size mismatch的问题,首先要了解到[a x b], [c x d]都代表什么m1 is [a x b] which is [batch size x in features]m2 is [c x d] which is [in features x out features]了解了这个以后,我们可以清楚的原创 2020-11-19 11:46:17 · 2540 阅读 · 2 评论 -
python对csv文件进行批量处理,修改csv某一列的数据并存储到新的csv文件中
导论本文中采用os库对文件进行批量导入, 采用csv库进行操作,主要是csv.read()和csv.write()本文解决了在新的csv文件中有空格的问题本文处理的是文本文档,采用‘wt’,‘rt’,不能使用‘wb’和‘rb’的二进制读取方式导入库import csvimport os创建文件夹source_path = './excel_data/'save_path = './csv_data/'if not os.path.exists(save_path): os原创 2020-07-13 22:23:20 · 25752 阅读 · 9 评论