stephanie

stephanie / pytorchjitcpu / 0.1.1

README.md

Overview

Demo of a PyTorch image object classifier trained on CIFAR-10. Identifies images as: 'plane', 'car', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck'.

Applicable Scenarios and Problems

Demonstrating hosting and deploying a serialized PyTorch model on Algorithmia.

Usage

Input

A single image path from data collections (or any of our data connectors that use the data api)

Output

Predicted class from possible labels listed above.

Examples

Input: "data://stephanie/pytorch_test/sample_plane1.jpg"

Output: "Predicted class is: ['plane']"