search

foamliu / Deep-Image-Matting-PyTorch

Deep Image Matting implementation in PyTorch
MIT License
817 stars 185 forks source link

Pre-Trained model can be reused for "Mobile-Image-Matting"? #34

Closed InternetMaster1 closed 4 years ago

InternetMaster1 commented 4 years ago

Hi,

Will the pretrained model of this "Deep-Image-Matting-PyTorch" project work on the "Mobile-Image-Matting" library too?

In the Mobile-Image-Matting library,

1) In the Demo section of ReadMe, it gives the following link for the pretrained model, but it gives a 404 error

https://github.com/foamliu/Deep-Mobile-Matting/releases/download/v1.0/BEST_checkpoint.tar

2) In the Performance section of ReadMe, it gives following link in the table against my-stage0

https://github.com/foamliu/Deep-Image-Matting-v2/releases/download/v1.0/BEST_checkpoint.tar

But the above points to the large checkpoint file from the main "Deep-Image-Matting-v2" project.

Trying to evaluate on this second link give following error

AttributeError: Can't get attribute 'DIMModel' on <module 'models' from '/content/gdrive/My Drive/MobileImageMatting/Mobile-Image-Matting/models/__init__.py'>

Any help?