Unable to discriminate between two classes while using HRNet model #1183
Unanswered
sparshgarg23
asked this question in
General
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
So,I was experimenting with the HRNet model and I trained it on my dataset (which consists of 8 classes).
Dataset distribution is as follows
{'F_AF': 480, 'F_AS': 444, 'F_CA': 348, 'F_LA': 479, 'M_AF': 456, 'M_AS': 372, 'M_CA': 468, 'M_LA': 396
And I am using weighted sample in the dataloader with weights as follows
[0.0021, 0.0023, 0.0029, 0.0021, 0.0022, 0.0027, 0.0021, 0.0025
Although the model is able to perform well with a validation accuracy of approximately 83%,I observed that it ends up getting confused between two of the classes .
Some sample images belonging to the two classes are enclosed
I am resizing my images to 224x224,since HRnet is able to focus on high resolution features,I decided not to resize the images,but that ended up huriting the performance.
When I run the model on class 3 image,the predicted class comes out to be C1 with score 67%,the second top class is C3 with score 33%
the classifier which I am using on top of the HRNet features is shown below
Any reason as to why the model is getting confused between these 2 classes,and what steps can I take to ensure the model doesn't get confused between these two classes
Beta Was this translation helpful? Give feedback.
All reactions