SegNeXt学习记录（一）：配置环境测试代码

news2024/10/7 16:16:14

安装配置MMSegmentation环境

为了验证 MMSegmentation 和所需的环境是否安装正确，我们可以运行示例 python 代码来初始化分段器并推断演示图像：

from mmseg.apis import inference_segmentor, init_segmentor
import mmcv

config_file = 'configs/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes.py'
checkpoint_file = 'checkpoints/pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth'

# build the model from a config file and a checkpoint file
model = init_segmentor(config_file, checkpoint_file, device='cuda:0')

# test a single image and show the results
img = 'test.jpg'  # or img = mmcv.imread(img), which will only load it once
result = inference_segmentor(model, img)
# visualize the results in a new window
model.show_result(img, result, show=True)
# or save the visualization results to image files
# you can change the opacity of the painted segmentation map in (0, 1].
model.show_result(img, result, out_file='result.jpg', opacity=0.5)

在这里插入图片描述

安装MMCV官方

mmsegmention环境安装
Ubuntu20运行SegNeXt代码提取道路水体(一)——从mmsegmentation安装到测试代码环境配置全过程摸索

怎么把图片保存出来？
在SegNeXt/mmseg/models/segmentors/base.py里的show_result函数里加了out_file = 'demo/result_demo.png'
但是会报错
cv2.error: OpenCV(4.5.5) /io/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'

出现的问题

后面下载了ade数据集，刚开始运行

python tools/train.py configs/pspnet/pspnet_r50-d8_512x512_80k_ade20k.py

还能跑起来，但是不知道为什么就报错AttributeError: 'ConfigDict' object has no attribute 'log_level'
解决：AttributeError: ‘ConfigDict‘ object has no attribute ‘log_level‘

训练与测试

训练：

python tools/train.py configs/pspnet/pspnet_r50-d8_512x512_80k_ade20k.py

测试：

python tools/test.py work_dirs/pspnet_r50-d8_512x512_80k_ade20k/pspnet_r50-d8_512x512_80k_ade20k.py work_dirs/pspnet_r50-d8_512x512_80k_ade20k/iter_80000.pth --eval mIoU

想要把预测的图片生成出来：

python tools/test.py work_dirs/pspnet_r50-d8_512x512_80k_ade20k/pspnet_r50-d8_512x512_80k_ade20k.py work_dirs/pspnet_r50-d8_512x512_80k_ade20k/iter_80000.pth --show-dir work_results/ --eval mIoU

在这里插入图片描述

使用SegNeXt进行训练及预测

下载ImageNet预训练模型：
在这里插入图片描述
用相同的方法进行训练看看ade数据效果如何
把这个预训练放在这个路径下：pretrained/mscan_b.pth

在这里插入图片描述
修改数据的位置，放到这个路径下：/root/ADEChallengeData2016/images/training

出现错误：PermissionError: ADE20KDataset: [Errno 13] Permission denied: ‘/root/ADEChallengeData2016/images/training’

修改之后，

训练

python tools/train.py local_configs/segnext/base/segnext.base.512x512.ade.160k.py

预测

python tools/test.py work_dirs/segnext.base.512x512.ade.160k/segnext.base.512x512.ade.160k.py work_dirs/segnext.base.512x512.ade.160k/latest.pth --show-dir work_results_segnext/ --eval mIoU

结果：

在这里插入图片描述

代码详解

参考：
Ubuntu20运行SegNeXt代码提取道路水体(一)——从mmsegmentation安装到测试代码环境配置全过程摸索
Ubuntu20运行SegNeXt代码提取道路水体(二)——SegNeXt源代码安装到测试环境配置全过程摸索
MMSegmentation V0.27.0训练与推理自己的数据集（二）

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.coloradmin.cn/o/973766.html

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈，一经查实，立即删除！