search

axinc-ai / ailia-models

The collection of pre-trained, state-of-the-art AI models for ailia SDK
2.01k stars 320 forks source link

ADD PP-LiteSeg #800

Closed itsmeterada closed 2 years ago

itsmeterada commented 2 years ago

PP-LiteSeg: A Superior Real-Time Semantic Segmentation Model

https://github.com/PaddlePaddle/PaddleSeg/tree/develop/configs/pp_liteseg

Apache 2.0 license

ooe1123 commented 2 years ago

onnxエクスポート

[cityscapes]

python deploy/python/infer_onnx_trt.py \
    --config configs/pp_liteseg/pp_liteseg_stdc1_cityscapes_1024x512_scale1.0_160k.yml \
    --model_path model.pdparams \
    --width 1024 \
    --height 512

[camvid]

python deploy/python/infer_onnx_trt.py \
    --config configs/pp_liteseg/pp_liteseg_stdc1_camvid_960x720_10k.yml \
    --model_path model.pdparams \
    --width 1024 \
    --height 768