Skip to content

Commit

Permalink
added default handlers documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
harshbafna committed Mar 13, 2020
1 parent 087f6ed commit 4ab6f7c
Showing 1 changed file with 36 additions and 0 deletions.
36 changes: 36 additions & 0 deletions docs/default_handlers.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
# TorchServe default inference handlers

TorchServe provides following inference handlers out of box :

## image_classifier

* Description : Handles image classification models trained on imagenet dataset.
* Input : RGB image
* Output : Top 5 predictions and their respective probability of the image

For more details refer [examples](../examples/image_classifier)

## image_segmenter

* Description : Handles image segmentation models trained on imagenet dataset.
* Input : RGB image
* Output : Output shape as [CL H W], CL - number of classes, H - height and W - width.

For more details refer [examples](../examples/image_segmenter)

## object_detector

* Description : Handles object detection models.
* Input : RGB image
* Output : List of detected classes and bounding boxes respectively

For more details refer [examples](../examples/object_detector)

## text_classifier

* Description : Handles models trained on imagenet dataset.
* Input : text file
* Output : Class of input text

For more details refer [examples](../examples/text_classification)

0 comments on commit 4ab6f7c

Please sign in to comment.