您的位置:首页 > 财经 > 产业 > 东莞外贸公司网站建设_室内设计小白怎么入行_做网络营销推广_爱站网seo培训

东莞外贸公司网站建设_室内设计小白怎么入行_做网络营销推广_爱站网seo培训

2024/12/23 6:13:51 来源:https://blog.csdn.net/FL1623863129/article/details/142768972  浏览:    关键词:东莞外贸公司网站建设_室内设计小白怎么入行_做网络营销推广_爱站网seo培训
东莞外贸公司网站建设_室内设计小白怎么入行_做网络营销推广_爱站网seo培训

快速开始

为了让大家快速了解PaddleSeg,本文档使用一个简单示例进行演示。在实际业务中,建议大家根据实际情况进行调整适配。

在开始下面示例之前,请大家确保已经安装好PaddleSeg开发环境(安装说明)。

1 准备数据

本示例将使用视盘分割(optic disc segmentation)数据集,数据集的原始图像和分割效果图如下所示。

通过以下命令可以下载视盘分割数据集(下载链接),解压保存到PaddleSeg/data目录下。

cd PaddleSeg
mkdir data && cd data
wget https://paddleseg.bj.bcebos.com/dataset/optic_disc_seg.zip
unzip optic_disc_seg.zip
cd ..

2 准备配置文件

我们常用PaddleSeg配置化驱动方式进行开发,配置文件是模型训练、评估、预测和部署的关键。

配置文件中定义了分割模型、损失函数、训练超参、训练数据集、验证数据集等信息。

本示例使用的配置文件是:PaddleSeg/configs/quick_start/pp_liteseg_optic_disc_512x512_1k.yml

3 模型训练

在PaddleSeg根目录下,执行如下命令进行单机训练。

export CUDA_VISIBLE_DEVICES=0 # Linux下设置1张可用的卡
# set CUDA_VISIBLE_DEVICES=0 # Windows下设置1张可用的卡python tools/train.py --config configs/quick_start/pp_liteseg_optic_disc_512x512_1k.yml --save_interval 500 --do_eval --use_vdl --save_dir output

训练完成后,模型在验证集上的mIoU达到90.65%(数据可能变动),精度最高的模型权重保存在PaddleSeg/output/best_model目录。

4 模型评估

在PaddleSeg根目录下,执行如下命令,使用val.py脚本来评估模型的精度,即计算验证数据集的精度。

python tools/val.py --config configs/quick_start/pp_liteseg_optic_disc_512x512_1k.yml  --model_path output/best_model/model.pdparams

5 模型预测

在PaddleSeg根目录下,执行如下命令,使用predict.py脚本加载模型,对图像进行预测,并且保存预测结果。

python tools/predict.py --config configs/quick_start/pp_liteseg_optic_disc_512x512_1k.yml  --model_path output/best_model/model.pdparams --image_path data/optic_disc_seg/JPEGImages/H0002.jpg --save_dir output/result

预测完成,可以在PaddleSeg/output/result目录下查看预测结果,如下图。

6 后续

得到训练好的模型后,上述预测步骤可以得到预测结果。此外,我们还可以导出预测模型并部署,实现更快的推理速度,具体请参考PaddleSeg的详细文档。

 

参考文章:https://gitee.com/paddlepaddle/PaddleSeg/blob/release/2.9/docs/quick_start_cn.md

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com