sophgo / sophon-stream

Other
71 stars 11 forks source link

有没有max_batch大于1个的yolov5模型? #18

Open ijiami-01 opened 7 months ago

ijiami-01 commented 7 months ago

如题,我在加载模型的时候,发现max_batch只有1,那么理论上,应该一次推理只能接受一张图片,可我希望达到的效果是,多张(比如3张,或者可以自定义图片数量)图片送去推理,然后如果3张图片的推理结果都超过了阈值,那么才将检测框画出来。 image

yizhou-xu commented 7 months ago

下载的文件里有多batch的,也可以自己转。后面描述的逻辑可以自己实现