1准备config配置文件-数据集和pipeline(需要修改类别的颜色等等)
mmseg->datasets 存储了所有数据集的类别和配色等等,导入自己的一个py文件。
注册数据集config文件__init__(),这样才可以调用。(需要修改,要在__init__.py中添加自己的数据集类)
pipeline配置文件(需要修改)
config->base->datasets ->ZihaoDataset_pipeline.py文件。
这里需要修改的是数据集的类名和数据集路径
。以及还有其他一些参数。
2准备模型的config文件(需要修改算法的内容)
主要修改的是类别数目。路径为:work/20240116/mmsegmentation/Zihao-Configs。