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 Train only One Class? #267

Open
takielias opened this issue Nov 18, 2022 · 4 comments
Open

How to Train only One Class? #267

takielias opened this issue Nov 18, 2022 · 4 comments
Labels
question Further information is requested

Comments

@takielias
Copy link

takielias commented Nov 18, 2022

@andrewdalpino Thanks for your incredible creation. As A PHP lover, I have been expecting this type of project for a long time.

Though I'm new to ML, I need to train in only one label. I got the CIFAR-10 or the MNIST example. But I could not manage it. It's required multiple labels. How can I do it? Like I need to detect only pigeons in the image.

@raijyan
Copy link

raijyan commented Dec 24, 2022

In your example you still have two classes. "Contains pigeon", "Doesn't contain pigeon". You'd label your dataset as such.

@takielias
Copy link
Author

takielias commented Dec 25, 2022

In your example you still have two classes. "Contains pigeon", "Doesn't contain pigeon". You'd label your dataset as such.
@raijyan
Yes, I made 2 folders, 1. Pigeon, 2. None. But How should I define none? How to put everything else pigeon in none folder?

@andrewdalpino andrewdalpino added the question Further information is requested label May 27, 2023
@andrewdalpino
Copy link
Member

Hey thanks @takielias I'm just reading this now! It sounds like you want image segmentation? In other words, you want the model to draw a bounding box around the pigeons in an image? If so, then what you need is a multimodal (classifier + regressor) model that A. predicts the class of what's in the image and B. draws a bounding box around the object (usually this is done by predicting the top left and bottom right coordinates).

Unfortunately, Rubix ML does not offer multimodal prediction at the moment.

Thanks for your kind words, I appreciate your support.

@takielias
Copy link
Author

takielias commented May 27, 2023 via email

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

No branches or pull requests

3 participants