栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

MODNet抠图模型复现

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

MODNet抠图模型复现

MODNet抠图模型复现

前言复现步骤


前言

MODNet由香港城市大学和商汤科技于2020年11月首次提出,用于实时抠图任务。该模型十分优秀,不需要背景做输入即可实现抠图,且实时性非常高。但是官方代码未给出完整的训练示例,我在github上给出了一个完整的复现示例,在B站给出了一个粗略的原理讲解。

复现步骤
# 1. 下载代码并进入工作目录
git clone https://github.com/actboy/MODNet
cd 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.zip
unzip src/datasets/PPM-100.zip -d src/datasets

# 4. 训练模型
python src/trainer.py

# 5. 模型评估
python src/eval.py

# 6. 模型推理
python src/infer.py
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/739668.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号