0
$\begingroup$

I want to create a model which solve a multiclass classification problem. The main concept is:

every picture contain only one object the background is very simple all object is coming from the same object class (exampleÉ knives, hats, shoes, etc.) te model will learn and predict the name of the object. (example: the model learn all type of knives, and when it get an image it will tell us the name of the knife) To be clear her is an example, I have 50 types of knife, and the output of the model has to recognize the correct name of the knife. Knife name could be: Chef's Knife,Heavy Duty Utility Knife,Boning Knife, etc.

To solve this problem, I have started to use annotated, segmented (masked) images (Coocolike dataset) and MASK RCNN model.

As a first step, I got a prediction, but I really don't know if I'm on the right way.

My questions are:

For this problem, Mask RCNN could be the solution, or it is impossible to recognize a tiny difference between two objects from the same class (example Chef's Knife, Heavy Duty Utility Knife)? Have you ever seen almost the same problem/solution, with github repo? Thanks a lot G

ps. This forum is about DS, but I didn't find any place to share my idea with other people. If you know somewhere a good community/forum, please share me.

$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Browse other questions tagged or ask your own question.