即使批处理大小为32,我也有内存错误问题。你知道吗
我正在为yolo目标探测器设置一个输入管道。对于每个批,我都有一个image_batch
numpy数组(32,416,416,3)
。我设置了一个调整大小、翻转、噪声等的预测管道(注意:我没有预先设置seq(列表)并使用它。我正在更新每个批的列表)运行此操作时,出现错误:
Memory Error: np.zeros(output_shape, images.dtype) at Resize()
我怎样才能解决这个问题?你知道吗
Tags:
目前没有回答
相关问题 更多 >
编程相关推荐