Dockerfile

# ベースイメージ（CUDAなし環境向け。GPU使いたい場合は nvidia/cuda ベースに変更）
FROM python:3.9-slim

# 必要パッケージをインストール
RUN apt-get update && \
    apt-get install -y --no-install-recommends \
        git \
        ffmpeg \
        libsm6 \
        libxext6 \
        libgl1-mesa-glx \
        wget \
        && rm -rf /var/lib/apt/lists/*

# 作業ディレクトリ
WORKDIR /workspace

# AlphaPoseをクローン
RUN git clone https://github.com/MVIG-SJTU/AlphaPose.git

# 作業ディレクトリをAlphaPoseに移動
WORKDIR /workspace/AlphaPose

# 依存関係インストール
RUN pip install --upgrade pip && pip install -r requirements.txt

# モデルのダウンロードスクリプト実行
RUN python scripts/download_model.py
RUN mkdir -p pretrained_models

# app.py をコンテナにコピー（ホスト側に置いておく必要あり）
COPY app.py /workspace/AlphaPose/app.py

# ポート（Gradioで使用する）
EXPOSE 7860

# アプリ起動（Gradioなど）
CMD ["python", "app.py"]