benthic_classification / testmodel.py
danielhshi8224
hf repo fixed
9d06c04
raw
history blame contribute delete
314 Bytes
from transformers import AutoModelForImageClassification, AutoImageProcessor
m = AutoModelForImageClassification.from_pretrained("dshi01/convnext-tiny-224-7clss")
p = AutoImageProcessor.from_pretrained("dshi01/convnext-tiny-224-7clss")
print(m.config.model_type, m.config.num_labels) # expect convnext, 7 (if set)