imgaug批处理中的内存错误

2024-10-01 02:28:05 发布

您现在位置:Python中文网/ 问答频道 /正文

即使批处理大小为32,我也有内存错误问题。你知道吗

我正在为yolo目标探测器设置一个输入管道。对于每个批,我都有一个image_batchnumpy数组(32,416,416,3)。我设置了一个调整大小、翻转、噪声等的预测管道(注意:我没有预先设置seq(列表)并使用它。我正在更新每个批的列表)运行此操作时,出现错误:

Memory Error: np.zeros(output_shape, images.dtype) at Resize()

我怎样才能解决这个问题?你知道吗


Tags: 内存image目标列表管道错误npyolo