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

http://www.niftyadmin.cn/n/1030575.html

相关文章

基于paddlepaddle复现MODNet

基于paddlepaddle复现MODNet前言复现步骤前言 MODNet由香港城市大学和商汤科技于2020年11月首次提出,用于实时抠图任务 MODNet特性: 轻量级(light-weight )实时性高(real-time)预测时不需要额外的背景输入…

基于跨模态AI模型实现图像检索诗歌

基于跨模态AI模型实现图像检索诗歌前言一、原理二、流程三、代码前言 生活中我们,我们常常会听到这样的说法:“啊,此时此景,我想作首诗”,或者“啊,此时此景,我想吟诗一首。” 清华大学九歌系…

UWA Pipeline 功能详解|可视化配置自动测试

之前曾为大家介绍了UWA Pipeline的「私有云真机远程调试」功能,凭借“无需真机在手,一屏操作所有”的功能特性,测试团队可以在局域网内的PC网页端轻松实现对多台移动测试设备的手动操作。 今天我们为大家介绍UWA Pipeline 2.x 版本的另一个功…

资源打包关系依赖树

1)资源打包关系依赖树 ​2)Coroutines产生了大量GC的问题 3)手机端的Soft Shadow如何生效 4)Skybox的环境照明问题 这是第292篇UWA技术知识分享的推送。今天我们继续为大家精选了若干和开发、优化相关的问题,建议阅读时…

UWA真人真机测试权益升级说明

不少Unity开发者都了解UWA的真人真机测评服务,只需把集成了SDK的包提交到UWA官网,选定测试机型并提供用例,约一个工作日,报告即可完成。报告涵盖了CPU、GPU和内存等多方面大家关心的性能数据,并以丰富的优化建议和行业…

UWA Pipeline 功能详解|如何快速创建UWA服务

今天为大家详细介绍的是如何在一条UWA Pipeline内快速创建UWA服务,包括自动化测试、GOT Online测试、真人真机测试、本地资源检测服务,让UWA性能保障体系无缝融入到项目工业化之中,进一步解放团队的生产力。 1. 创建自动化测试服务 在之前的…

基于2022高考数学全国卷I概率题解题思路初步分析新冠病毒疫苗

基于2022高考数学全国卷I概率题解题思路初步分析新冠病毒疫苗1. 2022高考数学全国卷I概率题2. 卡方(χ2\chi^2χ2)检验原理回顾3. 解答2022高考数学全国卷I概率题4. 上海疫情分析5. 总结1. 2022高考数学全国卷I概率题 一医疗团队为研究某地的一种地方性疾病与当地居民的卫生习惯…

AssetBundle加载的TMP字体材质赋值失败

1)AssetBundle加载的TMP字体材质赋值失败 ​2)资源打包关系依赖树 3)Shader中设置ColorMask后,最终输出颜色的计算 4)Time.deltaTime和Time.unscaledDeltaTime值不一致 这是第293篇UWA技术知识分享的推送。今天我们继续…