Classes names
Hello,
I'm currently using your model ResNet34_finetuned_for_skin_diseases_by-abdlh from Hugging Face in a local application for skin disease classification.
The model loads and runs fine, but I couldn't find the class labels used during training. Would you mind sharing the list of labels (e.g. id2label or a classes.txt), or confirming which dataset/classes were used?
It would really help to properly interpret the model's predictions.
Thanks in advance and great work!
Best regards,
Jose Luis
Hello! I hope you're doing well. I used the DermNet dataset for this model and I believe it had 23 classes but I narrowed it down to 6 removing most of the classes causing confusion. This is a link to the GitHub repository for further help and details: https://github.com/abdllhh/dermavision
Regards,
Muhammad Abdullah
Thank you very much!
I will try to translate your classes to the disease names. :)
I'm also planing my own model and I'm looking for datasets, just starting! DermNet looks really interesting!!
Best Regards
Jose