# -*-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