Dockerfile

FROM ubuntu:20.04
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
git \
python3-pip \
ffmpeg \
libsm6 \
libxext6 \
curl \
unzip \
wget \
&& rm -rf /var/lib/apt/lists/*

RUN git clone https://github.com/stefanopini/simple-HRNet.git
WORKDIR /simple-HRNet
RUN pip3 install -r requirements.txt
WORKDIR simple-HRNet/models
RUN curl -c ./cookie -s -L "https://drive.google.com/uc?export=download&id=1assh2z-rVLZRSiNZ-2EE7MedRDqwodeP" > /dev/null && \
curl -Lb ./cookie "https://drive.google.com/uc?export=download&confirm=`awk '/download/ {print $NF}' ./cookie`&id=1assh2z-rVLZRSiNZ-2EE7MedRDqwodeP" -o pytorch.zip && \
unzip pytorch.zip -d ./pytorch && rm ./pytorch.zip && \
cd ../ && mkdir weights 

WORKDIR /simple-HRNet/weights 
RUN curl -c ./cookie -s -L "https://drive.google.com/uc?export=download&id=1zYC7go9EV0XaSlSBjMaiyE_4TcHc_S38" > /dev/null && \
curl -Lb ./cookie "https://drive.google.com/uc?export=download&confirm=`awk '/download/ {print $NF}' ./cookie`&id=1zYC7go9EV0XaSlSBjMaiyE_4TcHc_S38" -o pose_hrnet_w48_384x288.pth
RUN git submodule update --init --recursive
WORKDIR /simple-HRNet/models/detectors/yolo
RUN pip3 install -r requirements.txt
WORKDIR /simple-HRNet/models/detectors/yolo/weights
RUN chmod 777 download_weights.sh && ./download_weights.sh
WORKDIR /simple-HRNet