添加测试功能

This commit is contained in:
2024-09-25 15:19:45 +08:00
parent b8c1202957
commit 3189caf7aa
5 changed files with 133 additions and 3 deletions

View File

@@ -13,7 +13,6 @@ ENV PYTHONUNBUFFERED=1 \
# 安装依赖 # 安装依赖
COPY requirements.txt /app/requirements.txt COPY requirements.txt /app/requirements.txt
COPY packages /app/packages
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \ RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \
&& pip install --no-cache-dir -r requirements.txt && pip install --no-cache-dir -r requirements.txt

4
api_test.py Normal file
View File

@@ -0,0 +1,4 @@
import time
if __name__ == '__main__':
time.sleep(3600)

128
docker-compose-test.yml Normal file
View File

@@ -0,0 +1,128 @@
x-base:
&base_template
restart: always
x-project:
&project_template
<<: *base_template
image: fcb_photo_review:2.0.0
volumes:
- ./log:/app/log
x-paddle:
&paddle_template
<<: *base_template
image: fcb_paddle:0.0.1
volumes:
- ./log:/app/log
- ./model:/app/model
services:
ocr_api:
<<: *paddle_template
build:
context: ./services/paddle_services
container_name: ocr_api
hostname: ocr_api
command: [ '-w 1 ./services/ocr_api:app --bind 0.0.0.0:5001' ]
deploy:
resources:
reservations:
devices:
- device_ids: [ '0' ]
capabilities: [ 'gpu' ]
driver: 'nvidia'
# settlement_api:
# <<: *paddle_template
# container_name: settlement_api
# hostname: settlement_api
# command: [ '-w 5 ./services/settlement_api:app --bind 0.0.0.0:5002' ]
# deploy:
# resources:
# reservations:
# devices:
# - device_ids: [ '0' ]
# capabilities: [ 'gpu' ]
# driver: 'nvidia'
#
# discharge_api:
# <<: *paddle_template
# container_name: discharge_api
# hostname: discharge_api
# command: [ '-w 5 ./services/discharge_api:app --bind 0.0.0.0:5003' ]
# deploy:
# resources:
# reservations:
# devices:
# - device_ids: [ '0' ]
# capabilities: [ 'gpu' ]
# driver: 'nvidia'
#
# cost_api:
# <<: *paddle_template
# container_name: cost_api
# hostname: cost_api
# command: [ '-w 5 ./services/cost_api:app --bind 0.0.0.0:5004' ]
# deploy:
# resources:
# reservations:
# devices:
# - device_ids: [ '1' ]
# capabilities: [ 'gpu' ]
# driver: 'nvidia'
#
# clas_api:
# <<: *paddle_template
# container_name: clas_api
# hostname: clas_api
# command: [ '-w 5 ./services/clas_api:app --bind 0.0.0.0:5005' ]
# deploy:
# resources:
# reservations:
# devices:
# - device_ids: [ '0' ]
# capabilities: [ 'gpu' ]
# driver: 'nvidia'
#
# det_api:
# <<: *paddle_template
# container_name: det_api
# hostname: det_api
# command: [ '-w 5 ./services/det_api:app --bind 0.0.0.0:5006' ]
# deploy:
# resources:
# reservations:
# devices:
# - device_ids: [ '1' ]
# capabilities: [ 'gpu' ]
# driver: 'nvidia'
#
# dewarp_api:
# <<: *paddle_template
# container_name: dewarp_api
# hostname: dewarp_api
# command: [ '-w 5 ./services/dewarp_api:app --bind 0.0.0.0:5007' ]
# deploy:
# resources:
# reservations:
# devices:
# - device_ids: [ '0' ]
# capabilities: [ 'gpu' ]
# driver: 'nvidia'
api_test:
<<: *project_template
build:
context: .
container_name: api_test
hostname: api_test
depends_on:
- ocr_api
# - settlement_api
# - discharge_api
# - cost_api
# - clas_api
# - det_api
# - dewarp_api
command: [ 'api_test.py' ]

View File

@@ -12,8 +12,7 @@ ENV PYTHONUNBUFFERED=1 \
PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple
# 安装依赖 # 安装依赖
COPY requirements.txt /app/requirements.txt COPY ./services/paddle_services/requestments.txt /app/requirements.txt
COPY packages /app/packages
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \ RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \
&& pip install --no-cache-dir -r requirements.txt \ && pip install --no-cache-dir -r requirements.txt \
&& pip uninstall -y onnxruntime onnxruntime-gpu \ && pip uninstall -y onnxruntime onnxruntime-gpu \

Binary file not shown.

After

Width:  |  Height:  |  Size: 4.0 MiB