目标检测,即给定一张图片,识别出图片中的目标位置。 本例中使用的是pytorch框架的yolov5模型。可以直接使用训练好的开源模型,也可以基于开源模型的源码进行修改、重新训练,还可以基于算法、框架构建适合的模型。 输入数据:RGB格式图片,分辨率为 640×640,输入形状为(1,3,640,640),也即(batchsize,channel,height,width)。 输出数据:目标检测框的坐标值、置信度、类别。 输出数组需要经过一定后处理,才能显示为正常的图片。