import os
import matplotlib.pyplot as plt
import numpy as np
import scipy.misc
parser = argparse.ArgumentParser(description='Pre-processing DIV2K .jpeg images')
parser.add_argument('--pathFrom', default='../../../../datadrive/DIV2K',
help='directory of images to convert')
parser.add_argument('--pathTo', default='../../../../datadrive/DIV2K',
help='directory of images to save')
parser.add_argument('--split', default=False,
help='save individual images')
parser.add_argument('--select', default='',
help='select certain path')
args = parser.parse_args()
for (path, dirs, files) in os.walk(args.pathFrom):
print(path)
targetDir = path.replace(args.pathFrom, args.pathTo)
if len(args.select) > 0 and path.find(args.select) == -1:
continue
if not os.path.exists(targetDir):
os.mkdir(targetDir)
if len(dirs) == 0:
pack = {}
n = 0
for fileName in files:
(idx, ext) = os.path.splitext(fileName)
if ext == '.png':
png = sio.imread(os.path.join(path, fileName))
tensor = torch.Tensor(png.astype(float)).byte()
if args.split:
torch.save(tensor, os.path.join(targetDir, idx + '.pt'))
else:
pack[int(idx.split('x')[0])] = tensor
n += 1
if n % 100 == 0:
print('Converted ' + str(n) + ' images.')
if len(pack) > 0:
torch.save(pack, targetDir + '/pack.pt')
print('Saved pt binary.')
del packtools_png2npy.py
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
- 文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
推荐阅读更多精彩内容
- 这两者简直就是神器，根据我的理解和尝试，两者最大的区别是，后者更加easy一点，因为后者不需要设置步长等，只需要输...
- 问题描述： Error:Could not find com.android.tools.build:gradle...
- 解决方案： 在/android/app/build.gradle内编辑新增mavenLocal() google(...
