首页 > 解决方案 > skimage.transform.resize 比 scipy.misc.imresize 慢

问题描述

我正在训练一个神经网络,在训练时我在数据准备部分遇到了错误。这一行:

img = np.double(scipy.misc.imresize(img, [height, width, channels], interp='bilinear', mode = 'RGB'))

这行:

img2 = np.double(scipy.misc.imresize(img2, [height, width], interp='bilinear'))

显示错误,因为scipy.misc.imresize在较新版本的 scipy 中已弃用。我所做的是我使用skimage.transform.resize而不是scipy.misc.imresize,这是推荐的,因为:

img = np.double(resize(img, (height, width, channels)))

img2 = np.double(resize(img2, (height, width)))

有效。但问题是与scipy.misc.imresize相比,数据读取过程变得非常缓慢。我之前在旧版本的scipy.misc.imresize上测试过它。非常感谢这方面的任何帮助。谢谢!

PS 我正在 Google Colab 上训练模型。

标签: pythonscipyimage-resizingscikit-image

解决方案


您可以复制粘贴 imresize 的来源:

https://github.com/scipy/scipy/blob/v1.1.0/scipy/misc/pilutil.py#L513


推荐阅读