Jupyter 内核在加载图像进行训练时死机?如何保持宽高比不变?

Jupyter kernel dying while loading images for training? How to keep aspect ratio intact?

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

此外,如果有人能告诉我为什么会崩溃,因为我在终端上输入了 nvidia-smi,它只显示了一小部分已使用的 gpu 内存。我的大部分图片都是 1536x2048,其他的是 880x 左右。

这是内存问题。已解决。