Skip to content

Latest commit

 

History

History
35 lines (28 loc) · 3.08 KB

File metadata and controls

35 lines (28 loc) · 3.08 KB

TorchServe

TorchServe is a flexible and easy to use tool for serving PyTorch models.

Basic Features

Advanced 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