firse

Dockerfile

@@ -1,63 +1,63 @@
-FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
-ENV DEBIAN_FRONTEND=noninteractive
-RUN apt-get update && \
-    apt-get upgrade -y && \
-    apt-get install -y --no-install-recommends \
-    git \
-    zip \
-    unzip \
-    git-lfs \
-    wget \
-    curl \
-    # ffmpeg \
-    ffmpeg \
-    x264 \
-    # python build dependencies \
-    build-essential \
-    libssl-dev \
-    zlib1g-dev \
-    libbz2-dev \
-    libreadline-dev \
-    libsqlite3-dev \
-    libncursesw5-dev \
-    xz-utils \
-    tk-dev \
-    libxml2-dev \
-    libxmlsec1-dev \
-    libffi-dev \
-    liblzma-dev && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/*
+# FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
+# ENV DEBIAN_FRONTEND=noninteractive
 # RUN apt-get update && \
-#     apt-get install zip
-# RUN wget https://github.com/ChenyangQiQi/FateZero/releases/download/v0.0.1/style.zip && unzip style.zip 
-RUN useradd -m -u 1000 user
-USER user
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:${PATH}
-WORKDIR ${HOME}/app
+#     apt-get upgrade -y && \
+#     apt-get install -y --no-install-recommends \
+#     git \
+#     zip \
+#     unzip \
+#     git-lfs \
+#     wget \
+#     curl \
+#     # ffmpeg \
+#     ffmpeg \
+#     x264 \
+#     # python build dependencies \
+#     build-essential \
+#     libssl-dev \
+#     zlib1g-dev \
+#     libbz2-dev \
+#     libreadline-dev \
+#     libsqlite3-dev \
+#     libncursesw5-dev \
+#     xz-utils \
+#     tk-dev \
+#     libxml2-dev \
+#     libxmlsec1-dev \
+#     libffi-dev \
+#     liblzma-dev && \
+#     apt-get clean && \
+#     rm -rf /var/lib/apt/lists/*
+# # RUN apt-get update && \
+# #     apt-get install zip
+# # RUN wget https://github.com/ChenyangQiQi/FateZero/releases/download/v0.0.1/style.zip && unzip style.zip 
+# RUN useradd -m -u 1000 user
+# USER user
+# ENV HOME=/home/user \
+#     PATH=/home/user/.local/bin:${PATH}
+# WORKDIR ${HOME}/app
 
-RUN curl https://pyenv.run | bash
-ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
-ENV PYTHON_VERSION=3.10.9
-RUN pyenv install ${PYTHON_VERSION} && \
-    pyenv global ${PYTHON_VERSION} && \
-    pyenv rehash && \
-    pip install --no-cache-dir -U pip setuptools wheel
+# RUN curl https://pyenv.run | bash
+# ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
+# ENV PYTHON_VERSION=3.10.9
+# RUN pyenv install ${PYTHON_VERSION} && \
+#     pyenv global ${PYTHON_VERSION} && \
+#     pyenv rehash && \
+#     pip install --no-cache-dir -U pip setuptools wheel
 
-RUN pip install --no-cache-dir -U torch==1.13.1 torchvision==0.14.1
-COPY --chown=1000 requirements.txt /tmp/requirements.txt
-RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
+# RUN pip install --no-cache-dir -U torch==1.13.1 torchvision==0.14.1
+# COPY --chown=1000 requirements.txt /tmp/requirements.txt
+# RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
 
-COPY --chown=1000 . ${HOME}/app
-RUN ls -a
-# RUN cd ./FateZero/ckpt && bash download.sh
-# RUN cd ./FateZero/data && bash download.sh
-# ENV PYTHONPATH=${HOME}/app \
-#     PYTHONUNBUFFERED=1 \
-#     GRADIO_ALLOW_FLAGGING=never \
-#     GRADIO_NUM_PORTS=1 \
-#     GRADIO_SERVER_NAME=0.0.0.0 \
-#     GRADIO_THEME=huggingface \
-#     SYSTEM=spaces
-# CMD ["python", "app_fatezero.py"]
+# COPY --chown=1000 . ${HOME}/app
+# RUN ls -a
+# # RUN cd ./FateZero/ckpt && bash download.sh
+# # RUN cd ./FateZero/data && bash download.sh
+# # ENV PYTHONPATH=${HOME}/app \
+# #     PYTHONUNBUFFERED=1 \
+# #     GRADIO_ALLOW_FLAGGING=never \
+# #     GRADIO_NUM_PORTS=1 \
+# #     GRADIO_SERVER_NAME=0.0.0.0 \
+# #     GRADIO_THEME=huggingface \
+# #     SYSTEM=spaces
+# # CMD ["python", "app_fatezero.py"]

app.py

@@ -22,9 +22,9 @@ det_config = 'configs/faster_rcnn_r50_fpn_1x_coco.py'
 det_checkpoint = 'faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth'
 
 # initialize pose model
-pose_model = init_pose_model(pose_config, pose_checkpoint, device='cuda')
+pose_model = init_pose_model(pose_config, pose_checkpoint, device='cpu')
 # initialize detector
-det_model = init_detector(det_config, det_checkpoint, device='cuda')
+det_model = init_detector(det_config, det_checkpoint, device='cpu')
 
 def predict(img):
     mmdet_results = inference_detector(det_model, img)