init

2022-08-23 19:13:13 +02:00
commit 6604923b26
13 changed files with 901 additions and 0 deletions
@@ -0,0 +1,12 @@
+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