# 说明
代码fork from [MODNet官方代码](https://github.com/ZHKKKe/MODNet) 。本项目完善了数据准备、模型评价及模型训练相关代码
# 模型训练、评价、推理
```bash
# 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
```
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
代码fork from MODNet官方代码 。本项目完善了数据准备、模型评价及模型训练相关代码,因为数据集较大,里面提供了数据集的下载方式。配置环境简单,直接根据代码写word就行了。
资源推荐
资源详情
资源评论
收起资源包目录
python毕业设计基于MODNet的证件照生成软件【完整源码+数据集+训练好的模型权重.zip (20个子文件)
src
__init__.py 0B
eval.py 2KB
shape_predictor_68_face_landmarks.dat 95.08MB
infer.py 2KB
trainer.py 14KB
face.py 4KB
inference.py 6KB
setting.py 154B
align.py 3KB
requirements.txt 116B
models
__init__.py 0B
backbones
__init__.py 277B
mobilenetv2.py 5KB
wrapper.py 3KB
modnet.py 10KB
matting_dataset.py 6KB
pretrained
mobilenetv2_human_seg.ckpt 8.67MB
modnet_photographic_portrait_matting.ckpt 25.04MB
README.md 227B
README.md 641B
共 20 条
- 1
资源评论
- 躺平的菜狗2024-05-20这个资源总结的也太全面了吧,内容详实,对我帮助很大。
瞲_大河弯弯
- 粉丝: 1335
- 资源: 135
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功