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
  
        
    
Uh oh!
There was an error while loading. Please reload this page.
-
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