19 lines
488 B
Python
19 lines
488 B
Python
# -*-coding:utf-8-*-
|
||
def detector(img, model):
|
||
"""_summary_
|
||
|
||
Args:
|
||
img (str or numpy.ndarray): 图片路径或者像素矩阵
|
||
model (_type_): 预加载的模型
|
||
|
||
Returns:
|
||
rtn(numpy.ndarray): 渲染后的图片像素点
|
||
pred(numpy.ndarray): 检测而出的目标的坐标点、置信度和类别，shape=[n, 6]
|
||
"""
|
||
result = model(img)
|
||
|
||
return result.render()[0], result.pred[0].cpu().numpy()
|
||
|
||
if __name__ == '__main__':
|
||
pass
|
||
|