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 train my own dataset? #9

Closed LLL-YUE closed 1 month ago

LLL-YUE commented 1 year ago

Can I train my own dataset instead of any specific dataset?

NoelShin commented 1 year ago

Yes, you can do it. For that, you first need to generate pseudo-masks using mask_generator.py.

Kind regards, Noel