Image Classification
Build a production-ready image classifier from scratch.
What You Will Build
A CNN-based image classifier that can categorize images into custom categories with 95%+ accuracy.
Complete Code
import oneml
dataset = oneml.Dataset.load("./images")
train, test = dataset.split(0.8)
model = oneml.Model.from_preset("resnet50")
model.fit(train, epochs=10)
accuracy = model.evaluate(test)
print(f"Accuracy: {accuracy:.2%}")
model.deploy("my-classifier")