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

How to use on custom datasets？ #7

Closed HalvesChen closed 1 year ago

HalvesChen commented 1 year ago

Hello！ I want to use this method on custom datasets, but I don't know how to get pseudo mask. Could you give me help?

NoelShin commented 1 year ago

Dear @chendaozai

Thank you for reaching out and your interest in our work - you can check mask_generator.py to generate pseudo-masks with your images. You just need to set a list of image paths in the script and save the resulting file if necessary.

Kind regards, Noel