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

https://download.openmmlab.com/mmsegmentation/v0.5/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes/pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth

  

   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      from mmseg.apis import init_segmentor, inf
     
    
  
 

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/119428684