How to use ZhengPeng7/BiRefNet-DIS5K with BiRefNet:
# Option 1: use with transformers from transformers import AutoModelForImageSegmentation birefnet = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet-DIS5K", trust_remote_code=True)
# Option 2: use with BiRefNet # Install from https://github.com/ZhengPeng7/BiRefNet from models.birefnet import BiRefNet model = BiRefNet.from_pretrained("ZhengPeng7/BiRefNet-DIS5K")
想问一下这个模型对应的对应的Google drive里的pth权重是哪一个呢，云端硬盘上DIS有好多个权重
是有点多... 可以看后缀哈, 是BiRefNet-DIS-epoch_590.pth, 也就是DIS5K标准实验setting下得到的模型.
· Sign up or log in to comment
