1.はじめに

　本記事では、Podmanで物体検出モデルyolox動作環境のコンテナを作成して、コンテナをイメージファイルにして、更にそのファイルをtarファイルの方法を記述します。
　PodmanとはRedHat社が開発したコンテナ管理ツールです。PodmanのコマンドはDockerと互換性があります、詳しくは以下の記事などに詳しく記載されているので、そちらを参照してください。

構築環境としては、WSLとUbuntu 22.04.1 LTSでWindows11にUbuntu環境を用意し、そのUbuntu上で構築しました。
以下、Ubuntuでのコマンドは@Ubuntu、コンテナでのコマンドは@containerと上部に記載します。

2. Podmanでコンテナ作成

podman pullでコンテナイメージ（debian）を取得します。
podman runでコンテナイメージ（debian）をもとに新しいコンテナでプロセスを実行します。

# @Ubuntu
$ podman pull docker.io/debian
$ podman images
#output
"""
REPOSITORY                TAG         IMAGE ID      CREATED     SIZE
docker.io/library/debian  latest      07d9246c53a6  4 days ago  129 MB
"""

podman run -it --name=yolox docker.io/debian

一旦ここでコンテナから抜けます。

# @container
exit

3.コンテナに接続

コンテナ内でコマンドを実行してyoloxの環境を構築するために、podman exec を使ってコンテナに接続します。

#@Ubuntu
podman ps -a
#output
"""
CONTAINER ID  IMAGE                            COMMAND     CREATED      STATUS      PORTS       NAMES
293284215a81  docker.io/library/debian:latest  bash        8 hours ago  Created                 yolox
"""

podman start yolox
podman exec -it yolox /bin/sh

接続後、コンテナ内の環境を確認

#@container
cat /etc/os-release

#output
"""
PRETTY_NAME="Debian GNU/Linux 11 (bullseye)"
NAME="Debian GNU/Linux"
VERSION_ID="11"
VERSION="11 (bullseye)"
VERSION_CODENAME=bullseye
ID=debian
HOME_URL="https://www.debian.org/"
SUPPORT_URL="https://www.debian.org/support"
BUG_REPORT_URL="https://bugs.debian.org/"
"""

uname -a

#output
"""
Linux 293284215a81 5.10.102.1-microsoft-standard-WSL2 #1 SMP Wed Mar 2 00:30:59 UTC 2022 x86_64 GNU/Linux
"""

4.yoloxの動作環境を構築

4.1 パッケージインストール

パッケージ管理コマンドaptで必要なパッケージをインストール

#@container
apt update
apt upgrade
apt install build-essential libbz2-dev libffi-dev libssl-dev zlib1g-dev git wget pip
apt-get install protobuf-compiler

4.2 Cmakeのインストール

今回、cmakeの公式ページのレポジトリからcmake-3.15.2をインストール

#@container
cd home
wget https://github.com/Kitware/CMake/releases/download/v3.15.2/cmake-3.15.2.tar.gz
tar -zxvf cmake-3.15.2.tar.gz
cd cmake-3.15.2
./bootstrap
make
make install

cmakeのバージョンを確認

#@container
cmake --version

#output
"""
cmake version 3.15.2
CMake suite maintained and supported by Kitware (kitware.com/cmake).
"""

4.3 yoloxの動作環境の構築

まず、Pythonのバージョン確認

#@container
python3 -V

#output
"""
Python 3.9.2
"""

yoloxのソースをリポジトリから取得して、pipでライブラリをインストールする。

#@container
cd ..
git clone https://github.com/Megvii-BaseDetection/YOLOX
cd YOLOX
pip install -U pip
pip install -r requirements.txt
pip install -v -e.

5.yoloxのデモを動作

yoloxにあるデモコードを実行して、動作を確認する。

5.1 yoloxで用いるモデルのダウンロード

本記事では、yoloxの中で最も軽量なモデルyolox-nanoを用いて動作確認する。以下のコマンドでYOLOXのmodelフォルダ内にyolox_nano.pthをダウンロードする。

#@container
mkdir model
cd model
wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_nano.pth

5.2 デモの実行

リポジトリのREADMEにある通りにデモを実施します。実施後、正常に終了すればYOLOX_outputsフォルダに結果が出力されています。

#@container
cd ..
python3 tools/demo.py image -n yolox-nano -c model/yolox_nano.pth --path assets/dog.jpg --conf 0.25 --nms 0.45 --tsize 640 --save_result --device cpu

5.3 PyTorch モデルを ONNX モデル形式に変換

リポジトリのONNXRuntimeのREADMEにある通り、PyTorchモデルを ONNX モデル形式に変換します。

#@container
python3 tools/export_onnx.py --output-name yolox_nano.onnx -n yolox-nano -c model/yolox_nano.pth

5.4 ONNX形式のモデルでデモを実行

yolox_nano.onnxをmodelフォルダに移動させて、リポジトリのONNXRuntimeのREADMEにある通り、デモを実行します。実行後、正常に終了していれば以下のコマンドで指定したoutputs フォルダに画像が出力されます。

#@container
mv yolox_nano.onnx model
python3 demo/ONNXRuntime/onnx_inference.py -m model/yolox_nano.onnx -i assets/dog.jpg -o outputs -s 0.3 --input_shape 416,416

6. 実行中コンテナからイメージファイルの生成とtarファイルへの変換

6.1 実行中コンテナからイメージファイルの生成

実行中のコンテナIDをpodman psで確認します。CONTAINER IDは293284215a81ですね。

#@ubuntu
podman ps

#output
"""
CONTAINER ID  IMAGE                            COMMAND     CREATED      STATUS             PORTS       NAMES
293284215a81  docker.io/library/debian:latest  bash        9 hours ago  Up 34 minutes ago              yolox
"""

以下のコマンドで、実行中のコンテナをイメージファイルにします。
podman commit [CONTAINER ID] イメージ名

#@Ubuntu
podman commit 293284215a81 yolox_demo

#output
"""
Getting image source signatures
Copying blob 9c742cd6c7a5 skipped: already exists
Copying blob e54f37d0afcf done
Copying config 65dd28fca8 done
Writing manifest to image destination
Storing signatures
65dd28fca85a11445458cd8fb32ae2bda0eb64530be56721eb437ab1569b63f2
"""

podman imagesでイメージファイルを確認すると以下になります。REPOSITORYにlocalhost/yolox_demoが追加されているのがわかります。

#@Ubuntu
podman images

#output
"""
REPOSITORY                TAG         IMAGE ID      CREATED        SIZE
localhost/yolox_demo      latest      65dd28fca85a  6 minutes ago  4.41 GB
docker.io/library/debian  latest      07d9246c53a6  4 days ago     129 MB
"""

6.2 イメージファイルからtarファイルの生成

最後に作成したイメージファイルをtarファイルにします。このtarファイルを他のPCなどに転送して使用することができます。

#@Ubuntu
podman save localhost/yolox_demo >yolox_demo.tar

終了後、lsコマンドで確認すると、yolox_demo.tarが作成されていると思います。

7. おわりに

　本記事では、Podmanで物体検出モデルyolox動作環境のコンテナ（debian）を作成して、コンテナをイメージファイルにして、更にそのファイルをtarファイルの方法を記述しました。エンジニア歴2か月で不備・不足もあるかと思いますが、ご指摘いただければ幸いです。また、類似の動作を確認する方の参考になれば幸いです。