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
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
- 文/潘曉璐 我一進(jìn)店門忍宋，熙熙樓的掌柜王于貴愁眉苦臉地迎上來痕貌，“玉大人，你說我怎么就攤上這事糠排《娉恚” “怎么了？”我有些...
- 文/不壞的土叔 我叫張陵入宦，是天一觀的道長哺徊。 經(jīng)常有香客問我，道長乾闰，這世上最難降的妖魔是什么落追？ 我笑而不...
- 正文 為了忘掉前任，我火速辦了婚禮涯肩，結(jié)果婚禮上轿钠，老公的妹妹穿的比我還像新娘巢钓。我一直安慰自己，他們只是感情好疗垛，可當(dāng)我...
- 文/花漫 我一把揭開白布症汹。 她就那樣靜靜地躺著，像睡著了一般继谚。 火紅的嫁衣襯著肌膚如雪烈菌。 梳的紋絲不亂的頭發(fā)上，一...
- 文/蒼蘭香墨 我猛地睜開眼旺矾，長吁一口氣：“原來是場噩夢啊……” “哼！你這毒婦竟也來了夺克？” 一聲冷哼從身側(cè)響起箕宙，我...
- 正文 年R本政府宣布褐奥，位于F島的核電站咖耘，受9級特大地震影響，放射性物質(zhì)發(fā)生泄漏抖僵。R本人自食惡果不足惜，卻給世界環(huán)境...
- 文/蒙蒙 一缘揪、第九天 我趴在偏房一處隱蔽的房頂上張望耍群。 院中可真熱鬧义桂，春花似錦、人聲如沸蹈垢。這莊子的主人今日做“春日...
- 文/蒼蘭香墨 我抬頭看了看天上的太陽曹抬。三九已至溉瓶，卻和暖如春，著一層夾襖步出監(jiān)牢的瞬間谤民，已是汗流浹背堰酿。 一陣腳步聲響...
推薦閱讀更多精彩內(nèi)容
- 這兩者簡直就是神器碉咆，根據(jù)我的理解和嘗試抖韩，兩者最大的區(qū)別是，后者更加easy一點疫铜，因為后者不需要設(shè)置步長等茂浮，只需要輸...
- 問題描述： Error:Could not find com.android.tools.build:gradle...
- 解決方案： 在/android/app/build.gradle內(nèi)編輯新增mavenLocal() google(...
