add README

Signed-off-by: Xinyu Ge <xinyu.ge@zilliz.com>
2 years ago · 6a42d2ac20
1 changed files with 1 additions and 1 deletions
--- a/README.md
+++ b/README.md
@ -78,7 +78,7 @@ and a corresponding vector in numpy.ndarray given a video input data.
 	Input video data using local path in string or video frames in ndarray.


-**Returns**: *(list, list, Tensor)*
+**Returns**: *(list, list, torch.Tensor)*

 	A tuple of (labels, scores, features),
 which contains lists of predicted class names and corresponding scores.