使用多核的张量流对象检测并行处理多幅图像

时间:2019-10-11 08:17:36

标签: python-3.x python-multiprocessing object-detection

我是tensorflow的新手,没有涉及多处理。我正在使用自定义的对象检测模型来处理我的图像。该模型基于faster_rcnn_inception_v2_pets,并且我能够使用CPU和GPU来处理图像。当前,每个图像大约需要4秒钟来处理。当我检查任务管理器时,CPU和GPU没有得到充分利用。我发现,对于每个图像,一开始它使用CPU的时间大约是95%,当它移到GPU上时,结果几乎是瞬间出现的。我的问题是,是否可以通过使用多核处理来加快速度?

我一直在阅读多处理,并想知道是否有可能在CPU(示例8内核)中使用多处理来并行处理图像并将它们排队到GPU中。 (我想我读过某个地方,GPU一次只能处理1个图像。)由于CPU占用很长的时间来处理1个图像,而GPU却非常快,因此,如果以上可能,它将减少处理速度大量图像的时间。

谢谢。

0 个答案:

没有答案