Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to learn on other datasets #42

Closed
W-hary opened this issue Aug 15, 2023 · 3 comments
Closed

How to learn on other datasets #42

W-hary opened this issue Aug 15, 2023 · 3 comments

Comments

@W-hary
Copy link

W-hary commented Aug 15, 2023

Thanks for the author's contribution
We noticed that the author mentioned in Features that CutLER can learn unsupervised object detectors and instance segmentors solely on ImageNet-1K.
Whether CutLER can be used for our datasets, which is not included in the ImageNet-1K?

@xiaoxin05
Copy link

hello,i have some problems ,can you help me ?

@frank-xwang
Copy link
Collaborator

Hi, CutLER can be used for custom datasets! You can follow the pipeline of training CutLER on ImageNet-1k:

  1. Cut: Use MaskCut to provide segmentation masks for multiple instances of each image, following this section. And place the json file under "DETECTRON2_DATASETS/YOUR-DATASET/annotations/".
  2. Learn: after getting all pseudo labels on your datasets, you can follow this section to start the unsupervised model learning process. Note: 1) If your dataset is very small, it is recommended to load our ImageNet pretrained model weights and fine-tune the model on your dataset. 2) you need to follow detectron2 to understand how to train a model on custom datasets.
  3. Self-training stage is optional.

Hope it helps.

@frank-xwang
Copy link
Collaborator

Closing this issues. Please feel free to re-open it if you meet other issues.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants