Can we do predictions for sub-classes i.e. class within class?

VittalAB 0
Viewed 43

I need to classify an object into multiple classes. Normally we are familiar with multi-class classification with a single hierarchy, but in my case I have two levels of hierarchy. See the below images to get a clear picture of what I am talking about. so that if I want to classify an image, it should give me all three classes: one is the main class and two subclasses. For example, Class-1-1-1, or Class-1-2-3.

Hierarchy of class

Solution for any framework will work either Tensorflow or PyTorch. Thank You.

image-classification machine-learning conv-neural-network tensorflow pytorch
1 Answers

If you know your hierarchy tree, wouldn't it be ok for you to do multi-class classification on the leaves (the final classes), then check what are the parent classes in the tree, for a given prediction ?

PlainRavioli0
Related