以segformer为例,CV中显存不够的处理方式

方法一:降低batchsize

方法二:将opencv库改成PIL

How to train segmentation model with large dataset? · Issue #1775 · open-mmlab/mmsegmentation (github.com)

你可能感兴趣的:(目标识别/语义分割,计算机视觉)