CSAILVision / CSAILVision/semantic-segmentation-pytorch
Error while inferencing binary classification trained model (hrnet+c1)
Open
- Dominant language
- Python
- Stars
- 5.1k
- Forks
- 1.1k
- PR merge metrics
- No merged PRs in 30d
Description
Hi Team,
Greetings..!!
This is in regards to an inference error. I have trained the model as a binary classifier (object and non-object). I have made changes to unm_class to 2 in the yaml config file. The annotations were made keeping 1 as object and 2 as non-object.
The model is trained on hrnetv2 and c1, however I am getting the above error. (please find attached screenshot)
Request you to assist me or guide me to the right direction.
Regards,
Akash Bhiwgade
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.