forked from wffjwbbf/ComDesignProject
Update README.md
This commit is contained in:
parent
76697fa840
commit
ab64f31b08
|
|
@ -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型,后期处理可能涉及到类型转换。*
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue