search

JXingZhao / EGNet

EGNet:Edge Guidance Network for Salient Object Detection (ICCV 2019)
315 stars 63 forks source link

是不是输入图像尺寸过大会导致报错？ #13

Open StormArcher opened 5 years ago

StormArcher commented 5 years ago

是不是图像尺寸过大会导致报错？ RuntimeError: CUDA error: out of memory 唯独每次在这个图片(比较大)的时候报错， 注意到其他图片的尺寸长宽最大在400, 而这张图(1125，1340)，这个网络的输入图像是不是不能太大？ 或者说我增大我的GPU到 11G 1080Ti就可以运行了？？（现在用的显卡是GTX1070的）

JXingZhao commented 5 years ago

是的呢，图片太大显存会炸掉，你可以考虑把这种过大的图片resize。

在 2019-09-28 20:54:57，Archerzjc notifications@github.com 写道：

是不是图像尺寸过大会导致报错？ RuntimeError: CUDA error: out of memory 唯独每次在这个图片(比较大)的时候报错， 注意到其他图片的尺寸长宽最大在400, 而这张图(1125，1340)，这个网络的输入图像是不是不能太大？ 或者说我增大我的GPU到 11G 1080Ti就可以运行了？？（现在用的显卡是GTX1070的） — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or mute the thread.