包名修改

Dockerfile

@@ -1,10 +1,12 @@
-FROM python:3.7.16
+FROM  tensorflow/tensorflow:2.5.1
 
-RUN pip install gunicorn==20.1.0
-RUN pip install setuptools==46.1.3
+RUN apt-get install make g++ gcc
+RUN pip3 install gunicorn
+RUN pip3 install setuptools==46.1.3
 
 RUN mkdir -p /app
 WORKDIR /app
+
 COPY requirements.txt /app
 RUN pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/
 

Dockerfile_arm

@@ -1,16 +0,0 @@
-FROM nvcr.io/nvidia/l4t-tensorflow:r32.6.1-tf2.5-py3
-RUN apt-get install make g++ gcc
-RUN pip3 install gunicorn
-RUN pip3 install setuptools==46.1.3
-
-RUN mkdir -p /app
-WORKDIR /app
-
-COPY requirements.txt /app
-RUN pip3 install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/
-
-COPY ./app /app
-EXPOSE 5000
-CMD ["gunicorn", "--bind", ":5000", "server:app"]
-
-

build.sh

@@ -1,3 +1,3 @@
 # /usr/bash
 
-docker build --tag hpds-road-detection:1.0.0 .
+docker build --tag hpds-tunnel-detection:1.0.0 .

docker-compose.yml

@@ -1,13 +1,13 @@
 version: "3.6"
 services:
-  hpds-python-model:
-    container_name: hpds-road-detection-model
-    image: hpds-road-detection:1.0.0 
+  hpds-tunnel-detection-model:
+    container_name: hpds-tunnel-detection-model
+    image: hpds-tunnel-detection:1.0.0
     networks:
       - hpds-network
     restart: always
     ports:
-      - "8000:5000"
+      - "8001:5000"
     volumes:
       - /usr/local/cuda/lib64:/usr/local/cuda/lib64
 

requirements.txt

@@ -1,5 +1,4 @@
 Flask==1.1.2
-numpy==1.19.5
+numpy==1.18.4
 Pillow==7.1.2
 six==1.15.0
-tensorflow==2.5.1