#encoding=utf-8
import sys
import os
import cv2
def make_directory(root,name):
filepath = os.path.join(root,name)
if not os.path.exists(filepath):
os.mkdir(filepath)
return filepath
def main():
## 读取所有的图片
all_picture_path = []
picture_dir_path = "face"
for root,dirs,files in os.walk(picture_dir_path):
for f in files:
all_picture_path.append(os.path.join(root,f))
## 过滤得到拼图
all_concat_picture_path = []
for picture_path in all_picture_path:
if len(picture_path.split("__")) > 10:
all_concat_picture_path.append(picture_path)
## 取出每张拼图标记
i = 0
while True:
concat_picture_path = all_concat_picture_path[i]
items = concat_picture_path[:-4].split("__")
# picture_path = "__".join(items[:-11])+".jpg"
picture_path = items[0] + ".jpg"
picture_dir_name = os.path.dirname(picture_path)
picture_base_name = os.path.basename(picture_path)
concat_picture = cv2.imread(concat_picture_path)
print picture_path
picture = cv2.imread(picture_path)
print concat_picture_path
cv2.imshow("face-top10",concat_picture)
# cv2.imshow("face",picture)
k = cv2.waitKey(0)
print k
if k == 49:##按下1
new_dir = make_directory(".","result")
new_dir = make_directory(new_dir,items[-10])
new_path = os.path.join(new_dir,picture_base_name)
cv2.imwrite(new_path,picture)
i += 1
elif k == 50:##2
new_dir = make_directory(".","result")
new_dir = make_directory(new_dir,items[-9])
new_path = os.path.join(new_dir,picture_base_name)
cv2.imwrite(new_path,picture)
i += 1
elif k == 51:##3
new_dir = make_directory(".","result")
new_dir = make_directory(new_dir,items[-8])
new_path = os.path.join(new_dir,picture_base_name)
cv2.imwrite(new_path,picture)
i += 1
elif k == 52:##4
new_dir = make_directory(".","result")
new_dir = make_directory(new_dir,items[-7])
new_path = os.path.join(new_dir,picture_base_name)
cv2.imwrite(new_path,picture)
i += 1
elif k == 53:##5
new_dir = make_directory(".","result")
new_dir = make_directory(new_dir,items[-6])
new_path = os.path.join(new_dir,picture_base_name)
cv2.imwrite(new_path,picture)
i += 1
elif k == 54:##6
new_dir = make_directory(".","result")
new_dir = make_directory(new_dir,items[-5])
new_path = os.path.join(new_dir,picture_base_name)
cv2.imwrite(new_path,picture)
i += 1
elif k == 55:##7
new_dir = make_directory(".","result")
new_dir = make_directory(new_dir,items[-4])
new_path = os.path.join(new_dir,picture_base_name)
cv2.imwrite(new_path,picture)
i += 1
elif k == 56:##8
new_dir = make_directory(".","result")
new_dir = make_directory(new_dir,items[-3])
new_path = os.path.join(new_dir,picture_base_name)
cv2.imwrite(new_path,picture)
i += 1
elif k == 57:##9
new_dir = make_directory(".","result")
new_dir = make_directory(new_dir,items[-2])
new_path = os.path.join(new_dir,picture_base_name)
cv2.imwrite(new_path,picture)
i += 1
elif k == 48:##0 第十个
new_dir = make_directory(".","result")
new_dir = make_directory(new_dir,items[-1])
new_path = os.path.join(new_dir,picture_base_name)
cv2.imwrite(new_path,picture)
i += 1
elif k == 110:## n
new_dir = make_directory(".","result")
new_dir = make_directory(new_dir,"unknow")
new_path = os.path.join(new_dir,picture_base_name)
cv2.imwrite(new_path,picture)
i += 1
if __name__=="__main__":
main()
