Fix saliency detection for `patch_size != 16`

YangtaoWANG95 / TokenCut

(CVPR 2022) Pytorch implementation of "Self-supervised transformers for unsupervised object discovery using normalized cut"

MIT License

304 stars 34 forks source link

Fix saliency detection for `patch_size != 16` #10

Closed Callidior closed 2 years ago

Callidior commented 2 years ago

The unsupervised saliency detection code currently does not work with patch size 8 because the feature extractor is always instantiated with patch size 16.

This PR fixes that.