segformer 使用笔记

这个是语义分割,不是实例分割,也不是全景分割。

开源地址:

https://github.com/NVlabs/SegFormer

安装环境:

pip install torchvision==0.8.2
pip install timm==0.3.2
pip install mmcv-full==1.2.7
pip install opencv-python==4.5.1.48
cd SegFormer && pip install -e . --user

报错:

pretrained/mit_b1.pth is not a checkpoint file

原因:pth路径写错了,改准确就行了。

预训练:

下载地址:

https://github.com/open-mmlab/mmsegmentation/issues/120

你可能感兴趣的:(pytorch知识宝典,python)