merge master and revise Dockerfile a bit.

XiSHEN0220 · Apr 18, 2024 · 0348fe2 · 0348fe2
2 parents d545c5c + b5a4e7c
commit 0348fe2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -10,3 +10,8 @@ RUN apt-get update && \
 RUN apt-get install -y libopencv-dev
 
 RUN pip install -r requirements.txt
+
+RUN cd /app/model/pretrained && \
+    bash download_model.sh
+RUN cd /app/data && \
+    bash Brueghel_detail.sh
diff --git a/Dockerfile_cuda10 b/Dockerfile_cuda10
@@ -5,4 +5,9 @@ COPY . .
 # to avoid the error "libgthread-2.0.so.0: cannot open", we need opencv itself.
 RUN apt-get update && \
     apt-get install -y libopencv-dev
-RUN bash requirement.sh
+RUN pip install -r requirements_cuda10.txt
+
+RUN cd /app/model/pretrained && \
+    bash download_model.sh
+RUN cd /app/data && \
+    bash Brueghel_detail.sh