File size: 652 Bytes
8568f38 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | #!/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()
|