YOLOv5提高检测速度:使用detect.py进行多进程推理
需求
我们的项目有一个需求,就是10s之内得检测完一张图片。但实际上程序可能需要的时间接近40s,所以就有了一个多线程的需求。所以下面测试我将通过跑3次相同路径下图片的耗时进行比较。使用的GPU设备是RTX3060。
原始推理
我先对他的main函数做了一些简单的修改。
def main(path):
opt = parse_opt(path)
check_requirements(ROOT / 'requirements.txt', exclude=('tensorboard', 'thop'))
return run(**vars(opt))
if __name__ == '__main__':
s_t = time.time