In [13]:
food101 = '/home/han/Downloads/food-101/images/'
import os
with open('./food101_calories.txt') as f:
calories = {
x.upper(): float(y) for x, y in map(str.split, f.readlines())
}
outdir = 'food101'
import scipy.misc
counter = 0
result = {}
for folder in os.listdir(food101):
cur_dir = os.path.join(food101, folder)
cal = calories[folder.replace('_', '+').upper()]
for fname in os.listdir(cur_dir):
cur_file = os.path.join(cur_dir, fname)
img = scipy.misc.imread(cur_file)
resized = scipy.misc.imresize(img, (256, 256))
new_file_name = os.path.join(outdir, str(counter) + '.jpg')
scipy.misc.imsave(new_file_name, resized)
counter += 1
result[new_file_name] = cal
In [15]:
with open('food101/calories.txt', 'w') as f:
for name, c in result.items():
print(name, c, file=f)