-
Notifications
You must be signed in to change notification settings - Fork 18
/
Dockerfile
103 lines (81 loc) · 2.79 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
# ----------------------------------- Base ----------------------------------- #
FROM python:3.8-slim as base
RUN apt-get update -qq \
&& apt-get install -y --no-install-recommends \
curl \
&& apt-get autoremove -y \
&& apt-get clean -y
# ---------------------------------- Builder --------------------------------- #
FROM base as builder
RUN apt-get update -qq && \
apt-get install -y --no-install-recommends \
git \
build-essential \
python3-dev \
libffi-dev \
libssl-dev \
# For PIL
zlib1g \
# For opencv
libsm6 \
libxext6 \
libxrender-dev \
# Tesseract
tesseract-ocr \
libtesseract-dev \
&& apt-get autoremove -y \
&& apt-get clean -y
# Install Poetry & ensure it is in $PATH
# RUN curl -sSL https://raw.githubusercontent.com/sdispater/poetry/master/get-poetry.py | POETRY_PREVIEW=1 python
RUN curl -sSL https://raw.githubusercontent.com/python-poetry/poetry/master/get-poetry.py | python
ENV PATH "/root/.poetry/bin:/opt/venv/bin:${PATH}"
# Copy deps information
COPY pyproject.toml poetry.lock /app/
# Copy spacy install script
COPY scripts/download-spacy-model.sh /app/
# Instals deps
RUN python -m venv /opt/venv && \
. /opt/venv/bin/activate && \
cd /app && \
pip install --no-cache-dir -U pip setuptools && \
poetry install --no-dev --no-root --no-interaction && \
# Download spacy model
bash download-spacy-model.sh && \
rm -rf dist *.egg-info
# ---------------------------------- Models ---------------------------------- #
FROM base as models
# Download OCR model
# RUN mkdir -p ./root/.keras-ocr
RUN mkdir -p ./root/.keras-ocr && ( \
cd /root/.keras-ocr && \
curl -L -o craft_mlt_25k.h5 https://github.com/faustomorales/keras-ocr/releases/download/v0.8.4/craft_mlt_25k.h5 && \
curl -L -o crnn_kurapan.h5 https://github.com/faustomorales/keras-ocr/releases/download/v0.8.4/crnn_kurapan.h5 \
)
# Download COCO Resnet model
COPY scripts/download-resnet-model.sh .
RUN bash download-resnet-model.sh
# Download EPIC-KITCHENS baseline model and noun classes
COPY scripts/download-epic-kitchens-model.sh .
RUN bash download-epic-kitchens-model.sh .
# ---------------------------------- Runner ---------------------------------- #
FROM base as runner
RUN apt-get update -qq && \
apt-get install -y --no-install-recommends \
ffmpeg \
libsm6 \
libxext6 \
&& apt-get autoremove -y \
&& apt-get clean -y
COPY --from=builder /opt/venv /opt/venv
COPY --from=models /root/.keras-ocr /root/.keras-ocr
COPY --from=models /data/coco_resnet /app/data/coco_resnet
COPY --from=models /data/epic_kitchens /app/data/epic_kitchens
COPY . app/
# Add the VirtualEnv to $PATH
ENV PATH="/opt/venv/bin:$PATH"
RUN . /opt/venv/bin/activate
WORKDIR /app
ENV PYTHONPATH=./ayesaac
# Add Entrypoint
RUN chmod +x ./scripts/wait-for-it.sh ./scripts/docker-entrypoint.sh
ENTRYPOINT ["./scripts/docker-entrypoint.sh"]