快速開(kāi)始

為了讓大家快速了解PaddleSeg，本文檔使用一個(gè)簡(jiǎn)單示例進(jìn)行演示。在實(shí)際業(yè)務(wù)中，建議大家根據(jù)實(shí)際情況進(jìn)行調(diào)整適配。

在開(kāi)始下面示例之前，請(qǐng)大家確保已經(jīng)安裝好PaddleSeg開(kāi)發(fā)環(huán)境（安裝說(shuō)明）。

1 準(zhǔn)備數(shù)據(jù)

本示例將使用視盤(pán)分割（optic disc segmentation）數(shù)據(jù)集，數(shù)據(jù)集的原始圖像和分割效果圖如下所示。

通過(guò)以下命令可以下載視盤(pán)分割數(shù)據(jù)集(下載鏈接），解壓保存到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 準(zhǔn)備配置文件

我們常用PaddleSeg配置化驅(qū)動(dòng)方式進(jìn)行開(kāi)發(fā)，配置文件是模型訓(xùn)練、評(píng)估、預(yù)測(cè)和部署的關(guān)鍵。

配置文件中定義了分割模型、損失函數(shù)、訓(xùn)練超參、訓(xùn)練數(shù)據(jù)集、驗(yàn)證數(shù)據(jù)集等信息。

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

3 模型訓(xùn)練

在PaddleSeg根目錄下，執(zhí)行如下命令進(jìn)行單機(jī)訓(xùn)練。

export CUDA_VISIBLE_DEVICES=0 # Linux下設(shè)置1張可用的卡
# set CUDA_VISIBLE_DEVICES=0 # Windows下設(shè)置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

訓(xùn)練完成后，模型在驗(yàn)證集上的mIoU達(dá)到90.65%(數(shù)據(jù)可能變動(dòng))，精度最高的模型權(quán)重保存在PaddleSeg/output/best_model目錄。

4 模型評(píng)估

在PaddleSeg根目錄下，執(zhí)行如下命令，使用val.py腳本來(lái)評(píng)估模型的精度，即計(jì)算驗(yàn)證數(shù)據(jù)集的精度。

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

5 模型預(yù)測(cè)

在PaddleSeg根目錄下，執(zhí)行如下命令，使用predict.py腳本加載模型，對(duì)圖像進(jìn)行預(yù)測(cè)，并且保存預(yù)測(cè)結(jié)果。

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

預(yù)測(cè)完成，可以在PaddleSeg/output/result目錄下查看預(yù)測(cè)結(jié)果，如下圖。

6 后續(xù)

得到訓(xùn)練好的模型后，上述預(yù)測(cè)步驟可以得到預(yù)測(cè)結(jié)果。此外，我們還可以導(dǎo)出預(yù)測(cè)模型并部署，實(shí)現(xiàn)更快的推理速度，具體請(qǐng)參考PaddleSeg的詳細(xì)文檔。

?

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