vggface2 / Dockerfile
jezzadebate's picture
Initial commit
b5d999a
raw
history blame contribute delete
362 Bytes
FROM python:3.9-slim-bullseye
RUN pip install --upgrade pip
RUN pip install facenet-pytorch
# Create working directory
RUN mkdir -p /usr/src/app
WORKDIR /usr/src/app
# Copy weights
ENV TORCH_HOME=/weights
COPY /weights /weights
# Copy source code
COPY face.py /usr/src/app
ENTRYPOINT ["python3", "/usr/src/app/face.py", "--input", "/in", "--output", "/out"]