本文展示了如何使用 Docker 容器中的 TensorFlow 目標檢測 API，通過網(wǎng)絡攝像頭執(zhí)行實時目標檢測，同時進行視頻后處理。作者使用的是 OpenCV 和 Python3 多進程和多線程庫。本文重點介紹了項目中出現(xiàn)的問題以及作者采用的解決方案。

完整代碼地址：https://github.com/lbeaucourt/Object-detection

用 YouTube 視頻進行視頻處理測試

動機

我是從這篇文章《Building a Real-Time Object Recognition App with Tensorflow and OpenCV》（https://towardsdatascience.com/building-a-real-time-object-recognition-app-with-tensorflow-and-opencv-b7a2b4ebdc32）開始探索實時目標檢測問題，這促使我研究 Python 多進程庫，使用這篇文章（https://www.pyimagesearch.com/2015/12/21/increasing-webcam-fps-with-python-and-opencv/）中介紹的方法提高每秒幀數(shù)（frames per second，F(xiàn)PS）。為了進一步加強項目的可移植性，我試著將自己的項目整合到 Docker 容器中。這一過程的主要困難在于處理流入和流出容器的視頻流。

此外，我還在項目中添加了視頻后處理功能，這一功能也使用了多進程，以減少視頻處理的時間（如果使用原始的 TensorFlow 目標檢測 API 處理視頻，會需要非常非常長的時間）。

在我的個人電腦上可以同時進行高性能的實時目標檢測和視頻后處理工作，該過程僅使用了 8GB 的 CPU。

用于數(shù)據(jù)科學的 Docker

鑒于大量文章對 TensorFlow 目標檢測 API 的實現(xiàn)進行了說明，因此此處不再贅述。作為一名數(shù)據(jù)科學家，我將展示如何在日常工作中使用 Docker。請注意，我用的是來自 Tensorflow 的經(jīng)典 ssd_mobilenet_v2_coco 模型。我在本地復制了模型（.pb 文件）和對應的標簽映射，以便后續(xù)個人模型的運行。

我相信現(xiàn)在使用 Docker 已經(jīng)是數(shù)據(jù)科學家最基礎(chǔ)的技能了。在數(shù)據(jù)科學和機器學習的世界中，每周都會發(fā)布許多新的算法、工具和程序，在個人電腦上安裝并測試它們很容易讓系統(tǒng)崩潰（親身經(jīng)歷?。?。為了防止這一悲慘事件的發(fā)生，我現(xiàn)在用 Docker 創(chuàng)建數(shù)據(jù)科學工作空間。

你可以在我的庫中找到該項目的相關(guān) Docker 文件。以下是我安裝 TensorFlow 目標檢測的方法（按照官方安裝指南進行）：

# Install tensorFlow RUN pip install -U tensorflow # Install tensorflow models object detection RUN git clone https://github.com/tensorflow/models /usr/local/lib/python3.5/dist-packages/tensorflow/models RUN apt-get install -y protobuf-compiler python-pil python-lxml python-tk #Set TF object detection available ENV PYTHONPATH "$PYTHONPATH:/usr/local/lib/python3.5/dist-packages/tensorflow/models/research:/usr/local/lib/python3.5/dist-packages/tensorflow/models/research/slim" RUN cd /usr/local/lib/python3.5/dist-packages/tensorflow/models/research && protoc object_detection/protos/*.proto --python_out=.

同樣，我還安裝了 OpenCV：

# Install OpenCV RUN git clone https://github.com/opencv/opencv.git /usr/local/src/opencv RUN cd /usr/local/src/opencv/ && mkdir build RUN cd /usr/local/src/opencv/build && cmake -D CMAKE_INSTALL_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local/ .. && make -j4 && make install

建立圖像會花幾分鐘的時間，但是之后用起來就會又快又容易。

實時目標檢測

首先我試著將目標檢測應用于網(wǎng)絡攝像頭視頻流?！禕uilding a Real-Time Object Recognition App with Tensorflow and OpenCV》完整地介紹了這項工作的主體部分。困難在于如何將網(wǎng)絡攝像頭視頻流傳送到 Docker 容器 中，并使用 X11 服務器恢復輸出流，使視頻得以顯示出來。

將視頻流傳送到容器中

使用 Linux 的話，設(shè)備在 /dev/ 目錄中，而且通?？梢宰鳛槲募M行操作。一般而言，你的筆記本電腦攝像頭是「0」設(shè)備。為了將視頻流傳送到 docker 容器中，要在運行 docker 圖像時使用設(shè)備參數(shù)：

docker run --device=/dev/video0

對 Mac 和 Windows 用戶而言，將網(wǎng)絡攝像頭視頻流傳送到容器中的方法就沒有 Linux 那么簡單了（盡管 Mac 是基于 Unix 的）。本文并未對此進行詳細敘述，但 Windows 用戶可以使用 Virtual Box 啟動 docker 容器來解決該問題。

從容器中恢復視頻流

解決這個問題時花了我一些時間（但解決方案仍舊不盡如人意）。我在 http://wiki.ros.org/docker/Tutorials/GUI 網(wǎng)頁發(fā)現(xiàn)了一些使用 Docker 圖形用戶界面的有用信息，尤其是將容器和主機的 X 服務器連接，以顯示視頻。

首先，你必須要放開 xhost 權(quán)限，這樣 docker 容器才能通過讀寫進 X11 unix socket 進行正確顯示。首先要讓 docker 獲取 X 服務器主機的權(quán)限（這并非最安全的方式）：

xhost +local:docker

在成功使用該項目后，再將控制權(quán)限改回默認值：

xhost -local:docker

創(chuàng)建兩個環(huán)境變量 XSOCK 和 XAUTH：

XSOCK=/tmp/.X11-unix XAUTH=/tmp/.docker.xauth

XSOCK 指 X11 Unix socket，XAUTH 指具備適當權(quán)限的 X 認證文件：

xauth nlist $DISPLAY | sed -e 's/^..../ffff/' | xauth -f $XAUTH nmerge -

最后，我們還要更新 docker 運行的命令行。我們發(fā)送 DISPLAY 環(huán)境變量，為 X11 Unix socket 和帶有環(huán)境變量 XAUTHORITY 的 X 認證文件安裝卷：

docker run -it --rm --device=/dev/video0 -e DISPLAY=$DISPLAY -v $XSOCK:$XSOCK -v $XAUTH:$XAUTH -e XAUTHORITY=$XAUTH

現(xiàn)在我們可以運行 docker 容器了，而它完成后是這樣的：

工作中的我和其他物體（因為害羞就不露臉了）。

盡管主機配置有 X 服務器，但我還是無法完全刪除我代碼中疑似錯誤的部分。OpenCV 需要通過調(diào)用 Python 腳本使用 cv2.imshow 函數(shù)進行「初始化」。我收到了以下錯誤信息：

The program 'frame' received an X Window System error.

然后，我可以調(diào)用 Python 主腳本（my-object-detection.py），視頻流也可以發(fā)送到主機的顯示器了。我對使用第一個 Python 腳本初始化 X11 系統(tǒng)的解決方法并不十分滿意，但是我尚未發(fā)現(xiàn)其他可以解決這一問題的辦法。

視頻處理

為了成功用網(wǎng)絡攝像頭實時運行目標檢測 API，我用了線程和多進程 Python 庫。線程用來讀取網(wǎng)絡攝像頭的視頻流，幀按隊列排列，等待一批 worker 進行處理（在這個過程中 TensorFlow 目標檢測仍在運行）。

就視頻處理而言，使用線程是不可能的，因為必須先讀取所有視頻幀，worker 才能對輸入隊列中的第一幀視頻應用目標檢測。當輸入隊列滿了時，后面讀取的視頻幀會丟失。也許使用大量 worker 和多個隊列可以解決這一問題（但會產(chǎn)生大量的計算損失）。

簡單隊列的另一個問題是，由于分析時間不斷變化，輸出隊列中的視頻幀無法以與輸入隊列相同的順序發(fā)布。

為了添加視頻處理功能，我刪除了讀取視頻幀的線程，而是通過以下代碼來讀取視頻幀：

while True: # Check input queue is not full if not input_q.full(): # Read frame and store in input queue ret, frame = vs.read() if ret: input_q.put((int(vs.get(cv2.CAP_PROP_POS_FRAMES)),frame))

如果輸入隊列未滿，則接下來會從視頻流中讀取下一個視頻幀，并將其放到隊列中去。否則輸入隊列中沒有視頻幀是不會進行任何處理的。

為了解決視頻幀順序的問題，我使用優(yōu)先級隊列作為第二輸出隊列：

1. 讀取視頻幀，并將視頻幀及其對應的編號一并放到輸入隊列中（實際上是將 Python 列表對象放到隊列中）。

2.然后，worker 從輸入隊列中取出視頻幀，對其進行處理后再將其放入第一個輸出隊列（仍帶有相關(guān)的視頻幀編號）。

while True: frame = input_q.get() frame_rgb = cv2.cvtColor(frame[1], cv2.COLOR_BGR2RGB) output_q.put((frame[0], detect_objects(frame_rgb, sess, detection_graph)))

3. 如果輸出隊列不為空，則提取視頻幀，并將視頻幀及其對應編號一起放入優(yōu)先級隊列，視頻編號即為優(yōu)先級編號。優(yōu)先級隊列的規(guī)模被設(shè)置為其他隊列的三倍。

# Check output queue is not empty if not output_q.empty(): # Recover treated frame in output queue and feed priority queue output_pq.put(output_q.get())

4. 最后，如果輸出優(yōu)先級隊列不為空，則取出優(yōu)先級最高（優(yōu)先級編號最?。┑囊曨l（這是標準優(yōu)先級隊列的運作）。如果優(yōu)先級編號與預期視頻幀編號一致，則將這一幀添加到輸出視頻流中（如果有需要的話將這一幀寫入視頻流），不一致的話則將這一幀放回優(yōu)先級隊列中。

# Check output priority queue is not empty if not output_pq.empty(): prior, output_frame = output_pq.get() if prior > countWriteFrame: output_pq.put((prior, output_frame)) else: countWriteFrame = countWriteFrame + 1 # Do something with your frame

要停止該進程，需要檢查所有的隊列是否為空，以及是否從該視頻流中提取出所有的視頻了。

if((not ret) & input_q.empty() & output_q.empty() & output_pq.empty()): break

總結(jié)

本文介紹了如何使用 docker 和 TensorFlow 實現(xiàn)實時目標檢測項項目。如上文所述，docker 是測試新數(shù)據(jù)科學工具最安全的方式，也是我們提供給客戶打包解決方案最安全的方式。本文還展示了如何使用《Building a Real-Time Object Recognition App with Tensorflow and OpenCV》中的原始 Python 腳本執(zhí)行多進程視頻處理。

原文鏈接：https://towardsdatascience.com/real-time-and-video-processing-object-detection-using-tensorflow-opencv-and-docker-2be1694726e5

責任編輯：xj

原文標題：教程 | 如何使用Docker、TensorFlow目標檢測API和OpenCV實現(xiàn)實時目標檢測和視頻處理

文章出處：【微信公眾號：新機器視覺】歡迎添加關(guān)注！文章轉(zhuǎn)載請注明出處。