python/example.py

#!/usr/bin/env python3
"""ResNet50 Age Classification Example."""
import sys
from resnet50_sdk import ResNet50Classifier

def main():
    if len(sys.argv) < 2:
        print("Usage: python example.py <image_path>")
        sys.exit(1)
    
    model_path = "../models/model.axmodel"  # relative to package
    image_path = sys.argv[1]
    
    classifier = ResNet50Classifier(model_path)
    results = classifier.classify(image_path, top_k=5)
    
    print(f"Image: {image_path}")
    print("Top-5 predictions:")
    for i, r in enumerate(results):
        print(f"  {i+1}. {r['label']}: {r['confidence']:.4f}")

if __name__ == "__main__":
    main()