13 lines
392 B
Docker
13 lines
392 B
Docker
FROM python:3.10
|
|
|
|
RUN pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
|
|
RUN pip install --user 'git+https://github.com/facebookresearch/detectron2.git'
|
|
RUN apt-get update
|
|
RUN apt-get install ffmpeg libsm6 libxext6 -y
|
|
RUN pip install opencv-python==4.6.0.66 pymongo==4.2.0
|
|
|
|
COPY ../model_v1 output/
|
|
COPY predict.py .
|
|
|
|
ENTRYPOINT python predict.py
|