类库 › matteformer
webtoon

webtoon/matteformer

MatteFormer是基于Transformer的图像抠图模型,利用先验Token和PAST块提升效果。提供SOTA性能评估及预训练模型,适用于Composition-1k等数据集。需Python 3.8、PyTorch 1.10.1及CUDA环境,支持从Deep Image Matting数据集准备数据并进行训练推理。

技术栈

根目录 python

查看全部依赖 (8)

依赖

NumPy easydict opencv-python pprint timm toml torch torchvision

评论

Home - Wiki
Copyright © 2011-2026 iteam. Current version is 2.155.2. UTC+08:00, 2026-07-30 06:56
浙ICP备14020137号-1 $Map of visitor$