我想把mydataset输出为txt文件保存应该怎么写 point_data = MyDataset(split='train', data_root=data_root, num_point=num_point, test_line=test_line, block_size=block_size, sample_rate=sample_rate, transform=None) print('point data size:', point_data.__len__()) print('point data 0 shape:', point_data.__getitem__(0)[0].shape) print('point label 0 shape:', point_data.__getitem__(0)[1].shape)
时间: 2024-04-10 14:34:15 浏览: 103
要将`MyDataset`的数据保存到一个txt文件中,你可以使用以下代码:
```python
point_data = MyDataset(split='train', data_root=data_root, num_point=num_point, test_line=test_line,
block_size=block_size, sample_rate=sample_rate, transform=None)
print('point data size:', point_data.__len__())
print('point data 0 shape:', point_data.__getitem__(0)[0].shape)
print('point label 0 shape:', point_data.__getitem__(0)[1].shape)
# 打开文件
fout = open("output.txt", 'w')
for i in range(point_data.__len__()):
data = point_data.__getitem__(i)
features = data[0]
label = data[1]
# 将数据写入文件
fout.write('v %f %f %f %d %d %d\n' % (
features[0], features[1], features[2], label[0], label[1], label[2]))
# 关闭文件
fout.close()
```
这段代码假设`MyDataset`的`__getitem__`方法返回的是一个包含特征和标签的元组。它会遍历`point_data`中的每个样本,将特征和标签写入文件。请记得根据你的实际需求修改文件路径和数据索引。
阅读全文