plemeri / InSPyReNet

Official PyTorch implementation of Revisiting Image Pyramid Structure for High Resolution Salient Object Detection (ACCV 2022)
MIT License
321 stars 61 forks source link

Error when loading pre-trained models #1

Closed TDeVries closed 1 year ago

TDeVries commented 1 year ago

Great work! The results look very nice. I was hoping to try out the pretrained models but I am getting errors when trying to load the checkpoints.

RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

I was wondering if this error is only on my end, or if there is an issue with the checkpoint files somehow?

plemeri commented 1 year ago

Hello, Thank you for your interest. I wonder why the zip error came out because our checkpoint file extension is .pth. Can you check if the checkpoint file name is latest.pth and the file is located as snapshots/InSPyReNet_SwinB/latest.pth?

TDeVries commented 1 year ago

Yes, the filepath and filename are correct.

Ah, I think the problem may be that it didn't download the entire file for some reason. My checkpoint files are only about 100MB, but it looks like they should be closer to 350MB.

plemeri commented 1 year ago

Please let me know if you need other download link.

TDeVries commented 1 year ago

If you have any other link that would be very helpful! The existing ones keep timing out before I can get the full file for some reason.

plemeri commented 1 year ago

I just updated ReadMe by adding google drive (GDrive) links for all assets that we provide. Hope this works!

TDeVries commented 1 year ago

I can download the full files now, thank you! It looks like there might be some permissions block on the new massive SOD dataset one though.

TDeVries commented 1 year ago

Got around to testing this. Amazing results! Thanks again for your responsive help!