| giolekva | 550add7 | 2020-04-11 20:01:40 +0400 | [diff] [blame] | 1 | FROM python:3 |
| 2 | |
| 3 | RUN pip help |
| 4 | RUN pip install torch --no-cache-dir |
| 5 | RUN pip install torchvision --no-cache-dir |
| 6 | RUN pip install facenet-pytorch --no-cache-dir |
| 7 | RUN pip install opencv-python-headless --no-cache-dir |
| 8 | RUN pip install matplotlib --no-cache-dir |
| 9 | |
| giolekva | b0b7f00 | 2020-04-16 16:05:40 +0400 | [diff] [blame] | 10 | # COPY . /face |
| 11 | WORKDIR /face |
| 12 | |
| 13 | # RUN rm -r -f output/* |
| 14 | CMD python face.py detect input output |
| 15 | # CMD python face.py classify output class |