Home PaddleSeg脚本可选参数
Post
Cancel

PaddleSeg脚本可选参数

参考PaddleSeg在GitHub的说明文档,做个记录。

PaddleSeg脚本可选参数

参考 PaddleSeg 说明文档

命令行FLAGS

FLAG用途支持脚本默认值备注
–cfg配置文件路径ALLNone 
–use_gpu是否使用GPU进行训练train/eval/visFalse 
–use_mpio是否使用多进程进行IO处理train/evalFalse打开该开关会占用一定量的CPU内存,但是可以提高训练速度。 NOTE: windows平台下不支持该功能, 建议使用自定义数据初次训练时不打开,打开会导致数据读取异常不可见。
–use_tb是否使用TensorBoard记录训练数据trainFalse 
–log_steps训练日志的打印周期(单位为step)train10 
–debug是否打印debug信息trainFalseIOU等指标涉及到混淆矩阵的计算,会降低训练速度
–tb_log_dirTensorBoard的日志路径trainNone 
–do_eval是否在保存模型时进行效果评估trainFalse 
–vis_dir保存可视化图片的路径vis“visual” 

训练

1
python3 pdseg/train.py --use_gpu --use_mpio --use_tb --tb_log_dir ./visual --do_eval --cfg ./configs/kitti_fscnn.yaml 
This post is licensed under CC BY 4.0 by the author.