densepose

docker-compose.yml

version: '2.3'
services:
  densepose:
    build: .
    runtime: nvidia
    volumes:
      - $HOME/densepose/data:/data
    ports:
      - 9999:8888

Dockerfile

# this Dockerfile is based on https://github.com/facebookresearch/DensePose/blob/master/docker/Dockerfile
FROM caffe2/caffe2:snapshot-py2-cuda9.0-cudnn7-ubuntu16.04

RUN mv /usr/local/caffe2 /usr/local/caffe2_build
ENV Caffe2_DIR /usr/local/caffe2_build

ENV PYTHONPATH /usr/local/caffe2_build:${PYTHONPATH}
ENV LD_LIBRARY_PATH /usr/local/caffe2_build/lib:${LD_LIBRARY_PATH}

RUN git clone https://github.com/facebookresearch/densepose /densepose
RUN pip install -r /densepose/requirements.txt

# Install the COCO API
RUN git clone https://github.com/cocodataset/cocoapi.git /cocoapi
WORKDIR /cocoapi/PythonAPI
RUN make install

# Install the DensePose
WORKDIR /densepose
RUN make
RUN make ops

# Download data to the container
RUN apt update
RUN apt install wget
WORKDIR /densepose/DensePoseData
RUN chmod +x *.sh
RUN ./get_DensePose_COCO.sh
RUN ./get_densepose_uv.sh
RUN ./get_eval_data.sh

WORKDIR /densepose
RUN pip install jupyterlab
EXPOSE 8888
CMD jupyter lab --ip=0.0.0.0 --port=8888 --no-browser --allow-root --NotebookApp.token='whatadensepose'