类库
› matteformer
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