import os
import csv
import argparse
import numpy as np
import scipy.misc
parser = argparse.ArgumentParser()
parser.add_argument('-f', '--file', required=True, help="path of the csv file")
parser.add_argument('-o', '--output', required=True, help="path of the output directory")
args = parser.parse_args()
w, h = 48, 48
image = np.zeros((h, w), dtype=np.uint8)
id = 1
with open(args.file, 'r') as csvfile:
datareader = csv.reader(csvfile, delimiter=',')
headers = next(datareader)
print (headers)
for row in datareader:
emotion = row[0]
pixels = list(map(int, row[1].split()))
usage = row[2]
pixels_array = np.asarray(pixels)
image = pixels_array.reshape(w, h)
image_folder = os.path.join(args.output, emotion)
if not os.path.exists(image_folder):
os.makedirs(image_folder)
image_file = os.path.join(image_folder , str(id) + '.jpg')
scipy.misc.imsave(image_file, image)
id += 1
if id % 100 == 0:
print('Processed {} images'.format(id))
print("Finished processing {} images".format(id))
2019-04-290
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
- 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
推荐阅读更多精彩内容
- 因为女儿想吃馄饨,早上把大宝和小宝送到学校。我就去买肉和馄饨皮。 买完肉回到家,我又忙着收拾一下卫生。就开始包馄饨...