From ab64f31b080e83f495483676149015e51392b1fe Mon Sep 17 00:00:00 2001 From: SeekingForJoy Date: Sun, 16 Oct 2022 00:31:51 +0800 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index f320593..e9234c8 100644 --- a/README.md +++ b/README.md @@ -20,7 +20,7 @@ * `detect(self,imgs:torch.Tensor) -> list` ->参数：`imgs`(类型为torch.Tensor)，大小应为(B,3,H,W),B为照片的张数，当为单张照片时为B==1即可。3时RGB三通道，W、H分别为图片的高和宽，**要求为64的倍数**。 +>参数：`imgs`(类型为torch.Tensor)，大小应为(B,3,H,W)。其中B为照片的张数，当为单张照片时为B==1即可。‘3’是RGB三通道，W、H分别为图片的高和宽，**要求为64的倍数**。 > >返回值:一个包含所有图片的bouding_box的列表，列表中每个元素对应每一张图片，其类型是二维的torch.Tensor；第二维是对应单张图片中的bouding_box;最后一维长度为6，分别对应(左上对角点横坐标`x1`，左上对角点纵坐标`y1`,右下角坐标横坐标`x2`，右下角纵坐标`y2`,可信度`conf`,类别`cls`)。*注意元素类型都为float型,后期处理可能涉及到类型转换。*