Jupyter 内核在加载用于训练的图像时死亡?如何保持纵横比不变?

问题描述

当我尝试加载没有目标大小的图像时,我的 jupyter 内核正在死亡。它恰好在 4% 后停止。大约有 12000 张图像。当我将 128,128 放入 target_size 时。它做得很好。我正在使用 tensorflow gpu 和 8gb 显卡。我的主要目的是保持图像的纵横比相同,所以我想避免调整大小,或者是否有任何方法只输入一个维度并根据纵横比自动获取一个维度?感谢所有输入,谢谢。

另外,如果有人可以让我知道为什么它会崩溃,因为我在终端上输入了 nvidia-smi 并且它只显示使用了一小部分 gpu 内存。我的大部分图片都是 1536x2048,其他的都是 880x。

解决方法

这是内存问题。解决了。​​