这个是语义分割,不是实例分割,也不是全景分割。
开源地址:
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路径写错了,改准确就行了。
预训练:
下载地址: