deeptangle/Dockerfile at main · kirkegaardlab/deeptangle · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
FROM python:3.8-slim-buster

ARG DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install wget unzip ffmpeg -y

COPY requirements.txt .

RUN ["pip3", "install", "--upgrade", "-r", "requirements.txt"]

WORKDIR /deeptangle
RUN ["wget", "https://sid.erda.dk/share_redirect/cEjIpG1yQl", "-O", "weights.zip"]
RUN ["unzip", "weights.zip"]

COPY celegans /deeptangle/celegans
COPY deeptangle /deeptangle/deeptangle
COPY examples /deeptangle/examples
COPY requirements.txt /deeptangle/requirements.txt
COPY setup.py /deeptangle/setup.py
COPY README.md /deeptangle/README.md


RUN ["pip3", "install", "-e", "."]

CMD ["python3", "examples/detect.py", "--model=weights/", "--input=/mnt/celegans_512.avi", "--output=/mnt/output.png", "--frame=100", "--correction_factor=1.2"]