ViTAMIn-O Models
Collection
Collection of various models associated with the ViTAMIn-O baseline model and specialized models for organoid differentiation outcome prediction. • 2 items • Updated
# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("ViTAMIn-O/ViTAMIn-O_base_model")
model = AutoModelForImageClassification.from_pretrained("ViTAMIn-O/ViTAMIn-O_base_model")Quick Links
ViTAMIn-O Generalist Model
This is the official baseline model, trained and used for inference in the corresponding paper:
ViTAMIn-O: Democratizing computer vision-based machine learning for stem cell research
Model Details
- Base Architecture:
microsoft/swin-large-patch4-window7-224 - Task Type:
Classification - Repository:
defefekt/ViTAMIn-O
Training Hyperparameters
- Seed:
42 - Epochs:
50 - Batch Size:
64
- Downloads last month
- 31
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="ViTAMIn-O/ViTAMIn-O_base_model") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")