foamliu/Deep-Image-Matting-PyTorch

pretrained tar archive is broken?

Closed this issue · 2 comments

i've downloaded the model but both 7z and tar reports unknown format.

any suggestions what to check?
i'm on windows, if it matters

you don't have to unzip the tarfile for loading weights, "torch.load" loads weights from tarfile itself.
like

checkpoint = 'BEST_checkpoint.tar'
checkpoint = torch.load(checkpoint,map_location='cpu')
model = checkpoint['model'].module
model = model.to(device)