From 6a42d2ac20545432dac5cba588f2e94aa791e0bc Mon Sep 17 00:00:00 2001 From: Xinyu Ge Date: Thu, 9 Jun 2022 17:35:00 +0800 Subject: [PATCH] add README Signed-off-by: Xinyu Ge --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index b7a0ca5..368ade0 100644 --- 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.