iamhmh/Skin-Lesion-Dataset-cls
Viewer • Updated • 25.3k • 39
How to use iamhmh/autotrain-isic-classification with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-classification", model="iamhmh/autotrain-isic-classification")
pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png") # Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("iamhmh/autotrain-isic-classification")
model = AutoModelForImageClassification.from_pretrained("iamhmh/autotrain-isic-classification")loss: 0.43574896454811096
f1_macro: 0.7652355174728005
f1_micro: 0.8454706927175843
f1_weighted: 0.8408555162880879
precision_macro: 0.8044349451647841
precision_micro: 0.8454706927175843
precision_weighted: 0.8433307863044311
recall_macro: 0.737854710508238
recall_micro: 0.8454706927175843
recall_weighted: 0.8454706927175843
accuracy: 0.8454706927175843
Base model
facebook/convnext-base-224-22k-1k