mmsegmentation 训练自制数据集

最新推荐文章于 2024-07-04 15:55:36 发布

GodBlessss

最新推荐文章于 2024-07-04 15:55:36 发布

阅读量577

点赞数

文章标签：深度学习机器学习 OpenMMLab

本文链接：https://blog.csdn.net/qq_41096807/article/details/126599431

版权

                    
                        
                    
                    以CHASE_DB1(眼底)数据集为参考，更换格式，运行tools/convert_datasets/BUSI.py
在mmseg/datasets/下定义对应数据集设置对应classes和palette，并在mmseg/datasets/init.py下进行添加
configs/base/datasets下更改数据路径；configs/base/models下更改class等参数；configs/base/schedules下更改all和pth保存epoch
train:
 python tools/train.py configs/unet/fcn_unet_s5-d16_128x128_40k_busi.py
Inference:
 python tools/test.py configs/unet/fcn_unet_s5-d16_128x128_40k_busi.py work_dirs/fcn_unet_s5-d16_128x128_40k_busi/iter_200000.pth --show-dir work_dirs/fcn_unet_s5-d16_128x128_40k_busi/imgs --eval mDice
图片展示:
 在mmseg/core/evaluation/class_names.py中添加对应数据集的classes和palatte, 修改demo/image_demo.py中的路径，运行

                