前言复现步骤
前言
MODNet由香港城市大学和商汤科技于2020年11月首次提出，用于实时抠图任务。该模型十分优秀，不需要背景做输入即可实现抠图，且实时性非常高。但是官方代码未给出完整的训练示例，我在github上给出了一个完整的复现示例，在B站给出了一个粗略的原理讲解。
复现步骤# 1、下载代码并进入工作目录git clone https://github.com/actboy/MODNetcd MODNet# 2、安装依赖pip install -r src/requirements.txt# 3、下载并解压数据集wget -c https://paddleseg.bj.bcebos.com/matting/datasets/PPM-100.zip -O src/datasets/PPM-100.zipunzip src/datasets/PPM-100.zip -d src/datasets# 4、训练模型python src/trainer.py# 5、模型评估python src/eval.py# 6、模型推理python src/infer.py