StartWorking

如何设置导出engine模型的batch呢？

7

转onnx 再转engine 默认batch为1，如何自定义？batch=1多路推理速度很慢

您好，大佬您的代码写的很好，我有几个问题想问您。我看Mainwindow只开了一个线程，相当于视频播放和yolo预测放在一个线程去做吗，如果用cpu预测的话不能满足实时性要求，播放会很卡（我用的cv2读的视频），那我目前的思路是两个：就是分两个线程去做，一个用于播放，一个用于预测，但是信号emit的图片我不太确定怎么写，因为2张图片要叠一起，我的展示框只有1个。还有一个思路就是yolo1秒钟预测几帧就行，这样预测和播放写一个线程就行了。是不是可以用第二种哈哈哈。 我看大佬yolo线程中的run是这么写的 batch = next(self.dataset) # next data print(batch) self.batch = batch # path, im, im0s, vid_cap, s = batch path, im, vid_cap, im0s = batch 请问这个batch是干嘛的呢，不是cv2读了cap 就可以model预测了吗