Skip to content

Latest commit

 

History

History
56 lines (45 loc) · 5.42 KB

README.md

File metadata and controls

56 lines (45 loc) · 5.42 KB

TorchServe

TorchServe is a performant, flexible and easy to use tool for serving PyTorch eager mode and torchscripted models.

Basic Features

Default Handlers

  • Image Classifier - This handler takes an image and returns the name of object in that image
  • Text Classifier - This handler takes a text (string) as input and returns the classification text based on the model vocabulary
  • Object Detector - This handler takes an image and returns list of detected classes and bounding boxes respectively
  • Image Segmenter- This handler takes an image and returns output shape as [CL H W], CL - number of classes, H - height and W - width

Examples

Advanced Features