【python】使用python脚本将CelebA中图片按照 list_attr_celeba.txt 中属性处理(删除、复制、移动)

103 篇文章 67 订阅 ¥69.90 ¥99.00
本文介绍了如何使用Python脚本`CelebAPickPic.py`根据CelebA数据集的`list_attr_celeba.txt`属性文件,执行对图片的删除、复制和移动操作。通过指定属性参数,可以筛选出具有特定属性的图片进行处理。支持的属性包括:5_o_Clock_Shadow到Young等四十种。
摘要由CSDN通过智能技术生成
1、目的

CelebA中的照片有四十种属性,参见:
【AI】CelebA数据介绍、下载及说明
根据需求从celebA中获取我们想要的图片,方法是将CelebA中图片按照 list_attr_celeba.txt 中属性执行删除、复制或移动操作。

命令格式:

python3 CelebAPickPic.py -a <attrFile> -i <imgDir> -d <delete|copy|move> -o <outDir>

参数说明:
-a :CelebA中属性说明文档list_attr_celeba.txt 的路径
-i :CelebA图片Img/img_align_celeba路径
-d :执行哪种操作,delete删除、copy复制、move移动
-o :如果是复制或移动,设置目标路径

执行脚本后,提示输入将要操作的属性名,以list_attr_celeba.txt 中列出的四十种属性,多个属性之间用空格分开。属性具体含义参见

如果您不想使用pandas库,也可以使用Python将DeepFashion数据集list_attr_img.txt文件转换为CSV文件。可以按照以下代码进行操作: ```python # 读取list_attr_img.txt文件 with open('list_attr_img.txt', 'r') as f: lines = f.readlines() # 处理每行数据 data = [] for line in lines[2:]: line = line.strip().split() img_path = line[0] attrs = line[1:] attrs = [int(x) for x in attrs] data.append([img_path] + attrs) # 将数据保存为CSV文件 with open('list_attr_img.csv', 'w') as f: # 写入列名 f.write('image_name,') f.write(','.join(lines[1].strip().split()[1:]) + '\n') # 写入数据 for line in data: line = [str(x) for x in line] f.write(','.join(line) + '\n') ``` 解释: 1. 首先使用open()函数读取list_attr_img.txt文件,将每行数据读取为一个字符串。 2. 对于每行数据,使用strip()函数去除空格和换行符,并使用split()函数将其拆分为图像路径和属性值。 3. 将属性值转换为整数类型。 4. 将图像路径和属性值组成一个列表,并添加到data列表。 5. 使用open()函数创建CSV文件,并使用write()函数将data列表的数据写入CSV文件。在写入CSV文件时,首先使用join()函数将列名转换为一个字符串,然后写入CSV文件;接着,使用join()函数将每行数据转换为一个字符串,并使用逗号分隔每个数据项。 这样,就可以使用Python将DeepFashion数据集list_attr_img.txt文件转换为CSV文件了。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

郭老二

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值