【mask2former】环境配置+cityscapes全景分割
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全景分割相关推荐
- cityscapes数据集_全景分割 UPSNet 源码分析 (1) - 数据格式
本系列文章针对的数据集是Cityscapes,后续会在Mapillary vista街道数据集上尝试训练出一版模型. 1. 数据集和标注文件 Cityscapes文件夹下 ├── annotation ...
- SWideRNet:全景分割新标杆!
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达 导读 本文是DeepLab系列作者"Liang-Chie ...
- 更快更强,谷歌提出SWideRNet:全景分割新标杆来啦!
作者丨Happy 编辑丨极市平台 导读 本文是DeepLab系列作者"Liang-Chieh Chen"大神在全景分割领域的又一力作.它在Wide-ResNet的基础上引入SE与& ...
- 全景分割这一年,端到端之路
机器之心原创 作者:朱艳芳 编辑:邱陆陆 图像分割(image segmentation)任务的定义是:根据某些规则将图片分成若干个特定的.具有独特性质的区域,并提出感兴趣目标的技术和过程. 目前图像 ...
- EfficientPS:目前排名第一的高效高精度全景分割算法
向大家推荐一个近期出现的全景分割算法 EfficientPS,在全景分割的 4 大数据集Cityscapes,.KITTI.Mapillary Vistas.IDD中测评精度全部位于榜首,且其语义分割 ...
- 何恺明组又出神作!最新论文提出全景分割新方法
↑ 点击上方[计算机视觉联盟]关注我们 来源:arXiv 作者:Alexander Kirillov et. al. 编辑:肖琴 Facebook人工智能实验室(FAIR)何恺明等人团队在arXiv发 ...
- 全景分割相关论文写作与准备笔记
第一次写论文,把一些自己在写论文时收集的资料和阅读心得等零零碎碎记录下来.(写作方向是全景分割.) CVPR经典论文introduction写法 VPS introduction: 全景分割的起源-& ...
- soloV2保姆级教程(含环境配置,训练自己的数据集,代码逻辑分析等。能踩得坑都踩了....)更新ing
序言 环境:ubuntu18.04 cuda11.1 显卡3050 pytorch1.9 代码链接: GitHub - aim-uofa/AdelaiDet: AdelaiDet is an ope ...
- 全景分割:CVPR2019论文解析
全景分割:CVPR2019论文解析 Panoptic Segmentation 论文链接: http://openaccess.thecvf.com/content_CVPR_2019/papers/ ...
最新文章
- python实现完全数
- Logback中如何自定义灵活的日志过滤规则
- post传参部分数据丢失
- Linux下文件权限查看并使用chomd修改文件权限
- 日常生活小技巧 -- UART 回环测试
- 如何在不亏本的情况下构建道德数据科学系统?
- ruby中正则表达式最小匹配与最大匹配
- 什么是分布式系统的CAP理论?
- 关于VS2017编译UE4项目,输出窗口有乱码的解决方案
- python设置窗体标题_Python3 tkinter基础 TK title 设置窗体的标题
- 银联支付接口常见错误总结
- 单片机c语言仿真,单片机c语言教程:C51表达式语句及仿真器
- html标签嵌套规则
- fastdb缩小初始生成文件
- Python索引设置
- Orcad 16.6中批量修改网络名
- unity测量模型尺寸
- iOS 依赖注入:Objection 和 Typhoon
- stay foolish, stay hungry
- 抱薪者说 | 从零开始 ,社区年会诞生记