Examples
Prediction
Single image, batch, and CSV export prediction examples.
Single image
from vyntri import Vyntri
model = Vyntri.load("model.vyntri")
# From file path
pred = model.predict("./photo.jpg")
print(f"Class: {pred.label}")
print(f"Confidence: {pred.confidence:.1%}")
# From PIL Image
from PIL import Image
img = Image.open("./photo.jpg")
pred = model.predict(img)
print(f"Class: {pred.label}")Batch prediction from folder
from vyntri import Vyntri
model = Vyntri.load("model.vyntri")
# Predict all images in a folder (recursive)
result = model.predict_batch("./images_folder")
for path, label, conf in zip(result.paths, result.labels, result.confidences):
print(f"{path}: {label} ({conf:.1%})")Batch prediction from list of paths
from vyntri import Vyntri
model = Vyntri.load("model.vyntri")
images = ["./photo1.jpg", "./photo2.jpg", "./photo3.jpg"]
result = model.predict_batch(images)
for label, conf in zip(result.labels, result.confidences):
print(f"{label}: {conf:.1%}")Export predictions to CSV
from vyntri import Vyntri
model = Vyntri.load("model.vyntri")
# Save predictions to CSV
result = model.predict_batch("./images_folder", out_path="predictions.csv")
print(f"{len(result.paths)} predictions saved to predictions.csv")