作者gigigigi (gigigigi)
看板Python
标题[问题] np.stack 错误问题
时间Sat Jun 16 18:03:15 2018
https://gist.github.com/shihyu/a56fb45dfb7781a48bcb7199cad25a0f
下面写成一行
train_data = np.stack([np.array(Image.open("./data/56_imitate_train_set/" +
'{0:05}'.format(int(row[0])) + ".jpg"))/255.0 for row in
csv.reader(traincsv)])
会出现 MemoryError, 下面错误讯息
Traceback (most recent call last):
File "test.py", line 22, in <module>
train_data = np.stack([np.array(Image.open("./data/56_imitate_train_set/"
+ '{0:05}'.format(int(row[0])) + ".jpg"))/255.0 for row in
csv.reader(traincsv)])
File
"/home/shihyu/anaconda3/lib/python3.6/site-packages/numpy/core/shape_base.py",
line 360, in stack
return _nx.concatenate(expanded_arrays, axis=axis, out=out)
MemoryError
但是把一行下面拆开来写是ok
for row in csv.reader(traincsv):
tmp = [np.array(Image.open("./data/56_imitate_train_set/" +
'{0:05}'.format(int(row[0])) + ".jpg"))/255.0]
train_data = np.stack(tmp)
请问这是什麽原因？
谢谢
--
※ 发信站: 批踢踢实业坊(ptt.cc), 来自: 175.181.180.179
※ 文章网址: https://webptt.com/cn.aspx?n=bbs/Python/M.1529143399.A.F63.html
1F：推 brahmsliu: 你把tmp 在每次回圈都重新定义了？所以不会爆 06/18 04:42