当前位置：首页 > 编程日记 > 正文

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

编程日记 2025-01-08 06:30:00

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 中列出的四十种属性，多个属性之间用空格分开。属性具体含义参见【AI】CelebA数据介绍、下载及说明
5_o_Clock_Shadow Arched_Eyebrows Attractive Bags_Under_Eyes Bald Bangs Big_Lips Big_Nose Black_Hair Blond_Hair Blurry Brown_Hair Bushy_Eyebrows Chubby Double_Chin Eyeglasses Goatee Gray_Hair Heavy_Makeup High_Cheekbones Male Mouth_Slightly_Open Mustache Narrow_Eyes No_Beard Oval_Face Pale_Skin Pointy_Nose Receding_Hairline Rosy_Cheeks Sideburns Smiling Straight_Hair Wavy_Hair Wearing_Earrings Wearing_Hat Wearing_Lipstick Wearing_Necklace Wearing_Necktie Young

2、CelebAPickPic.py

import os, getopt
import sys
import shutil
import stringdef getPath(argv):attrFile = ''imgDir = ''handle = ''outDir = ''try:opts, args = getopt.getopt(argv, "ha:i:d:o:", ["attr=","img=","handle=","out="])except getopt.GetoptError:print("<cmd> -a <attrFile> -i <imgDir> -d <delete|copy|move> -o <outDir>")sys.exit(2)for opt, arg in opts:if opt == '-h':print("<cmd> -a <attrFile> -i <imgDir> -d <delete|copy|move> -o <outDir>")sys.exit()elif opt in ("-a","--attr"):attrFile = argelif opt in ("-i", "--img"):imgDir = argelif opt in ("-d", "--handle"):if arg!="delete" and arg!="copy" and arg!="move" :print("<1cmd> -a <attrFile> -i <imgDir> -d <delete|copy|move> -o <outDir>")sys.exit()handle = argelif opt in ("-o", "--out"):outDir = argreturn attrFile, imgDir, handle, outDirdef handImg(img, handle, outDir):if handle=="delete":try:os.remove(img)except FileNotFoundError:print("FileNotFoundError : ", img)except :print("Other Error: ", img)elif handle == "copy":try:shutil.copy(img, outDir)except FileNotFoundError:print("FileNotFoundError : ", img)except :print("Other Error: ", img)elif handle == "move":try:shutil.move(img, outDir)except FileNotFoundError:print("FileNotFoundError : ", img)except :print("Other Error: ", img)if __name__ == '__main__':attrFile, imgDir, handle, outDir = getPath(sys.argv[1:])if not os.path.isfile(attrFile):print("属性文件不存在：", attrFile)sys.exit()if not os.path.isdir(imgDir):print("CeledA 图片路径不存在：", imgDir)sys.exit()if handle=="cpopy" or handle=="move":if not os.path.isdir(outDir):print("目标路径不存在：", outDir)sys.exit()print("attrFile = ", attrFile)print("imgDir = ", imgDir)print("handle = ", handle)print("outDir = ", outDir)fo = open(attrFile, "r")line = fo.readline()print('共有 {} 张图片'.format(line))line = fo.readline()attrList = line.split()print('属性列表：',attrList)isError = "err"attrHandle = []while isError=="err":print('输入需要处理的属性，多个属性之间用空格隔开：')attrHandle = input().split()print("将要处理的属性：", attrHandle)isError = "ok"for attr in attrHandle:if attr not in attrList:print("属性 {} 输入错误，请重新输入！".format(arrt))isError = "err"breakfor line in fo:tmp = line.split()for arrt in attrHandle:index = attrList.index(arrt)if tmp[index+1]=='1':handImg(str(os.path.join(imgDir, tmp[0])), handle, outDir)

https://www.dkcj.cn/info/35351.html

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

1、目的

2、CelebAPickPic.py

相关文章：

firefly 编译opencv3.3.1, CMake报错

AI时代，为何机器人公司无法盈利只能走向倒闭？

Google Objective-C Style Guide

ICPC 2019国际大学生程序设计竞赛，中国高校未能夺冠

完爆Facebook/GraphQL，APIJSON全方位对比解析(一)-基础功能

【AI】吴恩达斯坦福机器学习中文笔记汇总

【C++】C++命名空间重定向

宝宝都能看懂的机器学习世界

iOS LLDB调试命令(Low Lever Debug)

TCP通信速率与延时关系

十三、序列化和反序列化（部分转载）

【C++】google gflags详解

何恺明等人新作：效果超ResNet，利用NAS方法设计随机连接网络 | 技术头条

网络规划设计师考试命题模式持续在变你变不变

【C++】google glog详解

仿抖音注册界面制作

00后的AI开发者进阶之道：从入门到鏖战MIT编程大赛 | 人物志

js的全部替换函数replaceAll

【C++】Google Protocol Buffer（protobuf）详解（一）

python使用difflib对比文件示例

技术大佬们都推荐的vim学习指南来了，值得收藏！

NoSQl分类

手撕代码之七大常用排序算法 | 附完整代码

【C++】google gtest 详解

JS学习笔记之call、apply的用法

实验LVS+keepalived

Spark Streaming笔记整理（二）：案例、SSC、数据源与自定义Receiver

复旦邱锡鹏教授公布《神经网络与深度学习》，中文免费下载 | 极客头条

【Qt】信号和槽传递自定义结构体

Tomcat:Connection reset by peer: socket write error