Python Tensorflow:异步或线程

时间:2019-03-18 19:47:01

标签: python tensorflow websocket python-asyncio python-multithreading

我正在实现一个服务器,用于在“半真实”时间内使用tensorflow-gpu识别照片中的对象。它将在websocket连接上侦听新照片,然后将其放入空闲时运行的检测器列表中。使用asyncio或线程处理websocket侦听器和识别队列会更简单吗?

2 个答案:

答案 0 :(得分:0)

我不是线程处理/异步专家,但也许会更容易产生Kafka实例并拥有一段可以监听Kafka主题的代码?对于本主题,如果您已经将图像或图像路径存储在本地,则可以将其推送到图像。此外,使用consumer-groups可以免费获得负载平衡,因为它是Kafka的一部分。

答案 1 :(得分:0)

最终,我使用asyncio来处理websocket连接,将传入的图像排入队列。我使用了threading,它有一个线程将图像读入RAM,提取一些元数据,并将其排入对象检测器队列。在另一个线程中运行的检测器对图像进行了标记,并使标签在数据库处理程序中排队(还有另一个线程)。