用Python将deepfashion数据集的list_category_img.txt变成CSV文件
时间: 2024-03-24 13:41:15 浏览: 132
Python将list元素转存为CSV文件的实现
可以使用Python中的csv模块将deepfashion数据集的list_category_img.txt文件转换为CSV文件。
以下是示例代码:
```python
import csv
# 打开list_category_img.txt文件
with open('list_category_img.txt', 'r') as file:
lines = file.readlines()
# 分割每一行,将数据存储到列表中
data = []
for line in lines:
line = line.strip().split()
filename = line[0]
category_id = line[1]
data.append([filename, category_id])
# 将数据写入CSV文件
with open('list_category_img.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(data)
```
这段代码会将list_category_img.txt文件中的数据读入到一个列表中,然后将列表中的数据写入到list_category_img.csv文件中。你可以根据自己的需要修改文件名和路径。
阅读全文