|
|
@@ -30,10 +30,10 @@ RUN wget ftp://192.168.199.31/yiliv7_718.pt
|
|
|
|
|
|
RUN mkdir -p models
|
|
|
RUN cd models && wget ftp://192.168.199.31/yolov7p6_e6_300e/20240314/yolov7p6_e6_300e_coco.zip && unzip yolov7p6_e6_300e_coco.zip && rm yolov7p6_e6_300e_coco.zip
|
|
|
-# FIXME: replace with environment variables
|
|
|
-RUN sed -i 's/device = "cpu"/device = "gpu"/g' core/detectors/paddle_yolo/__init__.py && cat core/detectors/paddle_yolo/__init__.py
|
|
|
|
|
|
ADD . /workspace
|
|
|
+# FIXME: replace with environment variables
|
|
|
+RUN sed -i 's/device = "cpu"/device = "gpu"/g' core/detectors/paddle_yolo/__init__.py && cat core/detectors/paddle_yolo/__init__.py
|
|
|
EXPOSE 8080
|
|
|
|
|
|
|
