Pythonで画像ファイルの表示（Dockerコンテナ起動し、Terminalから実行）

解説

Dockerコンテナを起動し、起動したコンテナ上にいるTerminalからPythonプログラムを実行するときの、画像ファイルを表示する方法についてまとめます。
（JupyterLabサーバーをDockerコンテナを起動することで起動し、JupyterLab上のPythonプログラミングにおいて、画像ファイルを表示する方法は、「Pythonで画像ファイルの表示（Dockerコンテナ起動し、JupyterLabから実行）」にあります。）

画像ファイルを表示する方法は様々ありますが、以下の方法についてまとめます。

• OpenCVを用いて画像ファイルを読み込み、画像表示をする方法

機械学習開発用のコンテナの起動

「Dockerコンテナを実行してJupyterLabにアクセス」に従い、コンテナ起動スクリプトファイルを作成します。

「コンテナ上にいるTerminalから画像表示用のウィンドウを表示」できるようにするために、スクリプトファイルのコンテナ起動コマンドのオプションに以下を追加します。

--env DISPLAY=$DISPLAY

コンテナ起動スクリプトファイルの内容としては以下のようにします。
sudo docker run -it \ --rm \ --runtime nvidia \ --network host \ --volume ~/work:/work \ 　　　　　　　　--env DISPLAY=$DISPLAY \ nvcr.io/nvidia/l4t-ml:r35.2.1-py3

「コンテナ上にいるTerminalから画像表示用のウィンドウを表示」できるようにするために、コンテナを起動する前に以下のコマンドを実行します。（ローカルホストからのXサーバーへのアクセスを許可します。）

xhost +local:

以下のコマンドを実行し、「L4T ML」コンテナを起動します。

./docker_ml_run.sh

以降の操作は、起動したコンテナ上にいるTerminalで行ないます。

備考）
コンテナ起動コマンドのオプションに「--env DISPLAY=$DISPLAY」を追加しなかった場合、もしくは、コンテナ起動前に、「xhost +local:」を実行しなかった場合は、コンテナ上にいるTerminalからPythonプログラムを実行した際に、以下のようなメッセージが表示され、画像表示用のウィンドウが表示されません。
Unable to init server: Could not connect: Connection refused Traceback (most recent call last): File "showimage_cv2.py", line 7, in <module> cv2.imshow('Title you want', image) cv2.error: OpenCV(4.5.0) /opt/opencv/modules/highgui/src/window_gtk.cpp:624: error: (-2:Unspecified error) Can't initialize GTK backend in function 'cvInitSystem'

OpenCVを用いて画像ファイルを読み込み、画像表示をする方法

showimage_cv2.py
import cv2 # 画像ファイルの読み込み arrayImage = cv2.imread('baboon.jpg') # 画像の表示 cv2.imshow('Title you want', arrayImage) # キー入力待ち cv2.waitKey()

以下のコマンドを実行し、プログラムを実行します。
python3 showimage_cv2.py

実行結果例


画像ファイルの読み込み結果のimageオブジェクトのデータ型は、
numpy.ndarray
でした。（「print(type(arrayImage))」にて確認）

画像ファイルの読み込み結果のimageオブジェクトのNumPy配列の形状は、
(256, 256, 3)
でした。（256x256の画像を読み込んだ場合）（「print(arrayImage.shape)」にて確認）

コンテナの終了

コンテナを終了するには、起動したコンテナ上にいるTerminalで「exit」と入力します。

コンテナを終了したら以下のコマンドを実行します。（ローカルホストからのXサーバーへのアクセス許可を解除します。）

xhost -local:

参考ウェブサイト

• Unable to display images using Open CV when using docker container - Jetson & Embedded Systems / Jetson AGX Xavier - NVIDIA Developer Forums
• DockerでGUIアプリケーションを開く基本的な押さえどころ
• DockerでGUIを表示するときの仕組みについて - Qiita