search

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

salient mask selection #3

Closed ruoshuichengyang closed 1 year ago

ruoshuichengyang commented 2 years ago

Hello sir, if I want to train my own data, how can I get salient mask?How can I do offline spectral clustering to obtain candidate set?

NoelShin commented 1 year ago

Hello @ruoshuichengyang, I'm sorry for the very late response.

(1) I'll try to upload a script for computing a salient mask with your own images. (2) You can use "mask_generator.py" file (https://github.com/NoelShin/selfmask/blob/master/datasets/mask_generator.py) to obtain a candidate set.

I'll leave another comment here once done.

Sorry again for the late reply and thank you for your interest in our work.

NoelShin commented 1 year ago

A python script for generating pseudo-masks with your own images is added. Please use this file (https://github.com/NoelShin/selfmask/blob/master/mask_generator.py). You just need to set a list of image file paths.

Please let me know if you encounter any issues. :)