1. 配置mask2former环境

1.1 在conda下配置新环境:

conda create -n mask2former python=3.8
conda activate mask2former
pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html
conda install --channel https://conda.anaconda.org/Zimmf cudatoolkit=10.2
# pip install opencv-python
pip install -i https://pypi.doubanio.com/simple/ --trusted-host pypi.doubanio.com opencv-python

1.2 detectron2-main 安装

git clone git@github.com:facebookresearch/detectron2.git
下载文件夹解压主文件并打开:
cd detectron2-main
pip install -e . # 报错的话 执行这个 conda install -c conda-forge pycocotools
pip install "git+https://github.com/cocodataset/panopticapi.git"
pip install "git+https://github.com/mcordts/cityscapesScripts.git"
#linux环境下就要加"" , windows 就要去掉双引号

@@@@@如果遇到安装失败,参考pip install git+https:XXX 安装失败

python setup.py sdist
pip install cityscapesScripts-2.2.0.tar.gz

@@@@@@@@@@@@@@@@@这里指的是git+https://github.com/mcordts/cityscapesScripts.git安装失败后的补救办法。。。。。

cd ..

1.3 Mask2Former-main 安装

在Mask2Former-main 文件下:

cd Mask2Former-main
pip install -r requirements.txt
## 如果出现超时问题,socket.timeout: The read operation timed out
## pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple --default-timeout=100
## pip install -r requirements.txt -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
cd mask2former/modeling/pixel_decoder/ops
sh make.sh
# 数据集预处理需要的库
pip install opencv-python labelme
pip install setuptools==59.5.0 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

2. cityscape数据集设置



3. 训练测试

测试代码:

python ./demo/demo.py \
--config-file ./configs/cityscapes/panoptic-segmentation/maskformer2_R50_bs16_90k.yaml \--input "./input/000.jpg" \--confidence-threshold 0.5 --output "./output/000.jpg" \--opts MODEL.WEIGHTS ./model/mode_Cityscapes_Pano_Seg_R50.pkl

训练代码:
如果没有指定gpu,num-gpus 设置为1的话,就只会从0号gpu上面训练,如果需要指定,前面加上CUDA_VISIBLE_DEVICES=1

CUDA_VISIBLE_DEVICES=1 python train_net.py \--config-file ./configs/cityscapes/panoptic-segmentation/maskformer2_R50_bs16_90k.yaml \--num-gpus 1 \SOLVER.IMS_PER_BATCH 2 \SOLVER.BASE_LR 0.0025

中间训练遇到的问题:

1. AttributeError: module ‘distutils‘ has no attribute ‘version‘ 解决方案

“setuptools版本问题”,版本过高导致的问题
conda install setuptools==58.0.4 降低setuptools的版本

2.RuntimeError: Default process group has not been initialized, please make sure to call init_process_解决方案


这里的norm_cfg中,如果是多卡训练,采用SyncBN; 如果是单卡训练,将type修改为BN 即可。

然后训起来咯~~~~

【mask2former】环境配置+cityscapes全景分割相关推荐

  1. cityscapes数据集_全景分割 UPSNet 源码分析 (1) - 数据格式

    本系列文章针对的数据集是Cityscapes,后续会在Mapillary vista街道数据集上尝试训练出一版模型. 1. 数据集和标注文件 Cityscapes文件夹下 ├── annotation ...

  2. SWideRNet:全景分割新标杆!

    点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达 导读 本文是DeepLab系列作者"Liang-Chie ...

  3. 更快更强,谷歌提出SWideRNet:全景分割新标杆来啦!

    作者丨Happy 编辑丨极市平台 导读 本文是DeepLab系列作者"Liang-Chieh Chen"大神在全景分割领域的又一力作.它在Wide-ResNet的基础上引入SE与& ...

  4. 全景分割这一年,端到端之路

    机器之心原创 作者:朱艳芳 编辑:邱陆陆 图像分割(image segmentation)任务的定义是:根据某些规则将图片分成若干个特定的.具有独特性质的区域,并提出感兴趣目标的技术和过程. 目前图像 ...

  5. EfficientPS:目前排名第一的高效高精度全景分割算法

    向大家推荐一个近期出现的全景分割算法 EfficientPS,在全景分割的 4 大数据集Cityscapes,.KITTI.Mapillary Vistas.IDD中测评精度全部位于榜首,且其语义分割 ...

  6. 何恺明组又出神作!最新论文提出全景分割新方法

    ↑ 点击上方[计算机视觉联盟]关注我们 来源:arXiv 作者:Alexander Kirillov et. al. 编辑:肖琴 Facebook人工智能实验室(FAIR)何恺明等人团队在arXiv发 ...

  7. 全景分割相关论文写作与准备笔记

    第一次写论文,把一些自己在写论文时收集的资料和阅读心得等零零碎碎记录下来.(写作方向是全景分割.) CVPR经典论文introduction写法 VPS introduction: 全景分割的起源-& ...

  8. soloV2保姆级教程(含环境配置,训练自己的数据集,代码逻辑分析等。能踩得坑都踩了....)更新ing

    序言 环境:ubuntu18.04 cuda11.1 显卡3050  pytorch1.9 代码链接: GitHub - aim-uofa/AdelaiDet: AdelaiDet is an ope ...

  9. 全景分割:CVPR2019论文解析

    全景分割:CVPR2019论文解析 Panoptic Segmentation 论文链接: http://openaccess.thecvf.com/content_CVPR_2019/papers/ ...

最新文章

  1. python实现完全数
  2. Logback中如何自定义灵活的日志过滤规则
  3. post传参部分数据丢失
  4. Linux下文件权限查看并使用chomd修改文件权限
  5. 日常生活小技巧 -- UART 回环测试
  6. 如何在不亏本的情况下构建道德数据科学系统?
  7. ruby中正则表达式最小匹配与最大匹配
  8. 什么是分布式系统的CAP理论?
  9. 关于VS2017编译UE4项目,输出窗口有乱码的解决方案
  10. python设置窗体标题_Python3 tkinter基础 TK title 设置窗体的标题
  11. 银联支付接口常见错误总结
  12. 单片机c语言仿真,单片机c语言教程:C51表达式语句及仿真器
  13. html标签嵌套规则
  14. fastdb缩小初始生成文件
  15. Python索引设置
  16. Orcad 16.6中批量修改网络名
  17. unity测量模型尺寸
  18. iOS 依赖注入:Objection 和 Typhoon
  19. stay foolish, stay hungry
  20. 抱薪者说 | 从零开始 ,社区年会诞生记

热门文章

  1. 【报表设计】--层次坐标和父子格设置
  2. ANF框架小结、网络概念步骤详情及开发源码
  3. PT站点签到脚本,可挂青龙面板自动签到
  4. H5网页判断是否安装APP并调起APP
  5. 设置随机种子的函数: srand(种子);的重要性
  6. 小学学计算机应该学什么礼物,小学毕业礼物排行榜 运动学习两不耽误
  7. [iOS]如何向 appstore 查询已发布 APP 的信息?
  8. [深度文]YoloX部署、优化、训练相关
  9. 国家食品药品监督管理局 前有27家网站有资格向个人销售非处方药
  10. 【Unity3D实现自定义调色板】