0
点赞
收藏
分享

微信扫一扫

基于paddlepaddle复现MODNet

基于paddlepaddle复现MODNet


前言

MODNet由香港城市大学和商汤科技于2020年11月首次提出,用于实时抠图任务
MODNet特性:

  1. 轻量级(light-weight )
  2. 实时性高(real-time)
  3. 预测时不需要额外的背景输入(trimap-free)
  4. 准确度高(hight performance)
  5. 单模型(single model instead of a complex pipeline)
  6. 泛化能力强(better generalization ability)

代码特性:
paddlepaddle是百度开源的深度学习框架,国内唯一一个可与pytorch、tensorflow等媲美的深度学习框架
基于paddlepaddle实现的MODNet代码非常优雅、可读性强
包括完整的数据准备、模型训练、评估、预测及应用示例

复现步骤

 1. 下载源码
git clone https://github.com/PaddlePaddle/PaddleSeg.git

# 2. 安装环境依赖
conda create -n PaddleSeg python=3.7
conda activate PaddleSeg 
#GPU版本参考地址https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/windows-pip.html
python -m pip install paddlepaddle==2.2.2 -i https://mirror.baidu.com/pypi/simple
pip install scikit-image==0.19.1 scikit-learn==1.0.2 opencv-python==4.5.5.62 \
     PyYAML==6.0 filelock==3.5.0 visualdl==2.2.3

# 3. 设置环境变量PYTHONPATH
cd cd PaddleSeg/contrib/Matting
export PYTHONPATH=$PYTHONPATH:`pwd`/../..

# 4. 下载数据PPM-100数据集
mkdir -p data
wget -c https://paddleseg.bj.bcebos.com/matting/datasets/PPM-100.zip -O ./data/PPM-100.zip
unzip datasets/PPM-100.zip -d data

# 5. 训练模型
python train.py --config configs/modnet/modnet_mobilenetv2.yml --do_eval --use_vdl \
       --save_interval 5000 --num_workers 5 --save_dir output

# 6. 评估模型(通过百度提供的预训练模型)(在我的MacBook Pro上评价100张图片,约3.5分钟)
wget -c https://paddleseg.bj.bcebos.com/matting/models/modnet-mobilenetv2.pdparams \
       -O modnet-mobilenetv2.pdparams
python val.py  --config configs/modnet/modnet_mobilenetv2.yml \
       --model_path modnet-mobilenetv2.pdparams --save_dir ./output/results --save_results
         
# 7. 预测
python predict.py --config configs/modnet/modnet_mobilenetv2.yml \
    --model_path modnet-mobilenetv2.pdparams --image_path data/PPM-100/val/fg/ \
    --save_dir ./output/results
         
# 8. 背景更改应用(通过百度提供的预训练模型)
python bg_replace.py --config configs/modnet/modnet_mobilenetv2.yml \
    --model_path modnet-mobilenetv2.pdparams --image_path path/to/your/image  \
    --bg_path  path/to/your/background/image --save_dir ./output/results
举报

相关推荐

0 条评论