简体中文 | English
PaddleSeg 模型的推荐配置参数统一存放在 PaddleSeg/configs 下各个模型文件夹的 yaml 文件中。比如 ANN 的其中一个配置在 /PaddleSeg/configs/ann/ann_resnet50_os8_cityscapes_1024x512_80k.yml 中给出。如下图所示:
图中红色部分为骨干网络的预训练模型参数文件的存放位置。请注意:此处将直接以 https 链接形式下载我们提供的预训练模型参数。如果你在本地拥有骨干网络的预训练模型参数,请用其存放的绝对路径替换该 yaml 文件中
backbone
下的pretrained
。或者,你可以根据将要执行的train.py
所在的目录为该预训练模型参数的存放位置设置相对路径。
图中绿色部分为分割网络的预训练模型参数文件的存放位置。如果你在本地拥有分割网络的预训练模型参数,请用其存放的绝对路径或相对路径替换该 yaml 文件中的
pretrained
。
设置
epoch
的方式会受数据集大小的影响。因此PaddleSeg 按照iters
进行设置。
补充: 常见训练配置参数之间的关系
定义参数如下:
则有:
由于数据增强的配置要在yaml文件中进行指定,先介绍一下PaddleSeg中配置文件的基本知识。
Cityscapes是图像分割领域最常使用的数据集之一,因此Cityscapes上的一些常用配置已经给出。
PaddleSeg以
_base_
指定配置之间的继承关系:
_base_: '../_base_/cityscapes.yml'
_base_: '../_base_/cityscapes_1024x1024.yml'
transforms
所指定,由上到下依次加载。--batch_size 4
)覆盖 --config 内部的同名配置(如yaml中指定的batch_size: 8
)。如果你使用的是shape各不一致的自定义数据集,这可能是由于不得当的数据增强加载顺序引起的错误。
在Q3中,我们已经知道PaddleSeg的数据增强配置是按顺序加载的。
例如,
RandomRotation
会改变图片大小,如果在其他修正尺寸的增强配置之后设置它(如Crop、Resize),将导致图像尺寸的不一致。从而引发 DataLoader reader thread 错误。
因此,在开启训练之前,请参照Q3,仔细检查数据增强配置顺序。
目前在CityScapes上SOTA的模型可达到87%的mIoU。
详见: https://github.com/PaddlePaddle/PaddleSeg/tree/develop/contrib/CityscapesSOTA
best_model是在训练过程中验证对比得到的。因此需要在训练前开启选项
--do_eval
。
由于算力限制、各种不可抗力,模型可能未能一次训练完毕。
一个比较简单的办法是:把第一次生成的日志和第二次生成的日志中的内容拷贝到一个新的二进制文件中,然后读取。
我们将会在不久的将来在新版本中支持多个日志合并。
另外,如果是类似这种中断后继续训练的情况,可以在调用visualdl的时候指定日志名,这样就可以直接在指定的日志文件中继续续写了。
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》