pre-train weight for mask_generator.py

NoelShin / selfmask

[CVPRW'22] Unsupervised Salient Object Detection With Spectral Cluster Voting

https://www.robots.ox.ac.uk/~vgg/research/selfmask/

MIT License

59 stars 7 forks source link

pre-train weight for mask_generator.py #10

Open chinglun opened 8 months ago

chinglun commented 8 months ago

When i run mask_generator.py ,it need some pre-train weights like /users/gyungin/sos/networks/pretrained/moco_v2_800ep_pretrain.pth.tar Is it necessary? If so, how can i get it?

Thanks for your response!

NoelShin commented 7 months ago

Dear @baron0830

Apologies for the late reply and thanks for your interest in our work. Yes, they are necessary and you can download here:

For MoCov2: https://dl.fbaipublicfiles.com/moco/moco_checkpoints/moco_v2_800ep/moco_v2_800ep_pretrain.pth.tar For SwAV: https://dl.fbaipublicfiles.com/deepcluster/swav_800ep_pretrain.pth.tar

I believe in case of DINO, it'll download directly from a given URL in the code.

Hope this helps, Gyungin