Update README.md

This commit is contained in:
SeekingForJoy 2022-10-16 00:31:51 +08:00
parent 76697fa840
commit ab64f31b08
1 changed files with 1 additions and 1 deletions

View File

@ -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即可。3RGB三通道W、H分别为图片的高和宽**要求为64的倍数**。
>
>返回值:一个包含所有图片的bouding_box的列表列表中每个元素对应每一张图片其类型是二维的torch.Tensor第二维是对应单张图片中的bouding_box;最后一维长度为6分别对应(左上对角点横坐标`x1`,左上对角点纵坐标`y1`,右下角坐标横坐标`x2`,右下角纵坐标`y2`,可信度`conf`,类别`cls`)。*注意元素类型都为float型,后期处理可能涉及到类型转换。*