ML icon indicating copy to clipboard operation
ML copied to clipboard

How to Train only One Class?

Open takielias opened this issue 3 years ago • 4 comments

@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.

takielias avatar Nov 18 '22 15:11 takielias

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

raijyan avatar Dec 24 '22 19:12 raijyan

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?

takielias avatar Dec 25 '22 00:12 takielias

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.

andrewdalpino avatar May 27 '23 20:05 andrewdalpino

Yes, exactly. Waiting for the multi-model support of RubixML.

On Sun, May 28, 2023, 2:23 AM Andrew DalPino @.***> wrote:

Hey thanks @takielias https://github.com/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.

— Reply to this email directly, view it on GitHub https://github.com/RubixML/ML/issues/267#issuecomment-1565666230, or unsubscribe https://github.com/notifications/unsubscribe-auth/AJJBAZEAPHDANLDBW7A7QHTXIJPC5ANCNFSM6AAAAAASETLKF4 . You are receiving this because you were mentioned.Message ID: @.***>

takielias avatar May 27 '23 22:05 takielias