pamixsun
/

swinv2_tiny_for_glaucoma_classification

Image Classification

Model card Files Files and versions

pamixsun commited on Aug 9, 2023

Commit

8d4ab76

·

1 Parent(s): 462a897

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -71,7 +71,7 @@ inputs = processor(image, return_tensors="pt")
 with torch.no_grad():
     logits = model(**inputs).logits
-# model predicts one of the 1000 ImageNet classes
 predicted_label = logits.argmax(-1).item()
 print(model.config.id2label[predicted_label])

 with torch.no_grad():
     logits = model(**inputs).logits
+# model predicts either glaucoma or non-glaucoma.
 predicted_label = logits.argmax(-1).item()
 print(model.config.id2label[predicted_label])