Francesco/brain-tumor-m2pbp
Viewer • Updated • 9.9k • 200 • 3
How to use Aryan-401/tumor-detect-yolo-small with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("object-detection", model="Aryan-401/tumor-detect-yolo-small") # Load model directly
from transformers import AutoImageProcessor, AutoModelForObjectDetection
processor = AutoImageProcessor.from_pretrained("Aryan-401/tumor-detect-yolo-small")
model = AutoModelForObjectDetection.from_pretrained("Aryan-401/tumor-detect-yolo-small")Base model
hustvl/yolos-small