diff --git a/docker-compose.yml b/docker-compose.yml index 07984d7..64efd8c 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -35,44 +35,44 @@ services: capabilities: [ 'gpu' ] driver: 'nvidia' -# ie_settlement: -# <<: *paddle_template -# container_name: ie_settlement -# hostname: ie_settlement -# command: [ '-w', '1', 'ie_settlement:app', '--bind', '0.0.0.0:5002' ] -# deploy: -# resources: -# reservations: -# devices: -# - device_ids: [ '0' ] -# capabilities: [ 'gpu' ] -# driver: 'nvidia' -# -# ie_discharge: -# <<: *paddle_template -# container_name: ie_discharge -# hostname: ie_discharge -# command: [ '-w', '1', 'ie_discharge:app', '--bind', '0.0.0.0:5003' ] -# deploy: -# resources: -# reservations: -# devices: -# - device_ids: [ '0' ] -# capabilities: [ 'gpu' ] -# driver: 'nvidia' -# -# ie_cost: -# <<: *paddle_template -# container_name: ie_cost -# hostname: ie_cost -# command: [ '-w', '1', 'ie_cost:app', '--bind', '0.0.0.0:5004' ] -# deploy: -# resources: -# reservations: -# devices: -# - device_ids: [ '1' ] -# capabilities: [ 'gpu' ] -# driver: 'nvidia' + ie_settlement: + <<: *paddle_template + container_name: ie_settlement + hostname: ie_settlement + command: [ '-w', '1', 'ie_settlement:app', '--bind', '0.0.0.0:5002' ] + deploy: + resources: + reservations: + devices: + - device_ids: [ '0' ] + capabilities: [ 'gpu' ] + driver: 'nvidia' + + ie_discharge: + <<: *paddle_template + container_name: ie_discharge + hostname: ie_discharge + command: [ '-w', '1', 'ie_discharge:app', '--bind', '0.0.0.0:5003' ] + deploy: + resources: + reservations: + devices: + - device_ids: [ '0' ] + capabilities: [ 'gpu' ] + driver: 'nvidia' + + ie_cost: + <<: *paddle_template + container_name: ie_cost + hostname: ie_cost + command: [ '-w', '1', 'ie_cost:app', '--bind', '0.0.0.0:5004' ] + deploy: + resources: + reservations: + devices: + - device_ids: [ '1' ] + capabilities: [ 'gpu' ] + driver: 'nvidia' clas_orientation: <<: *paddle_template @@ -87,44 +87,61 @@ services: capabilities: [ 'gpu' ] driver: 'nvidia' -# det_book: -# <<: *paddle_template -# container_name: det_book -# hostname: det_book -# command: [ '-w', '1', 'det_book:app', '--bind', '0.0.0.0:5006' ] -# deploy: -# resources: -# reservations: -# devices: -# - device_ids: [ '1' ] -# capabilities: [ 'gpu' ] -# driver: 'nvidia' -# -# dewarp: -# <<: *paddle_template -# container_name: dewarp -# hostname: dewarp -# command: [ '-w', '1', 'dewarp:app', '--bind', '0.0.0.0:5007' ] -# deploy: -# resources: -# reservations: -# devices: -# - device_ids: [ '0' ] -# capabilities: [ 'gpu' ] -# driver: 'nvidia' -# -# clas_text: -# <<: *paddle_template -# container_name: clas_text -# hostname: clas_text -# command: [ '-w', '1', 'clas_text:app', '--bind', '0.0.0.0:5008' ] -# deploy: -# resources: -# reservations: -# devices: -# - device_ids: [ '1' ] -# capabilities: [ 'gpu' ] -# driver: 'nvidia' + det_book: + <<: *paddle_template + container_name: det_book + hostname: det_book + command: [ '-w', '1', 'det_book:app', '--bind', '0.0.0.0:5006' ] + deploy: + resources: + reservations: + devices: + - device_ids: [ '1' ] + capabilities: [ 'gpu' ] + driver: 'nvidia' + + dewarp: + <<: *paddle_template + container_name: dewarp + hostname: dewarp + command: [ '-w', '1', 'dewarp:app', '--bind', '0.0.0.0:5007' ] + deploy: + resources: + reservations: + devices: + - device_ids: [ '0' ] + capabilities: [ 'gpu' ] + driver: 'nvidia' + + clas_text: + <<: *paddle_template + container_name: clas_text + hostname: clas_text + command: [ '-w', '1', 'clas_text:app', '--bind', '0.0.0.0:5008' ] + deploy: + resources: + reservations: + devices: + - device_ids: [ '1' ] + capabilities: [ 'gpu' ] + driver: 'nvidia' + + photo_review_test_1: + <<: *project_template + build: + context: . + container_name: photo_review_test_1 + hostname: photo_review_test_1 + depends_on: + - ocr + - ie_settlement + - ie_discharge + - ie_cost + - clas_orientation + - det_book + - dewarp + - clas_text + command: [ 'photo_review.py', '--clean', 'True' ] # photo_review_1: # <<: *project_template @@ -175,15 +192,12 @@ services: # - photo_review_4 # command: [ 'photo_review.py' ] - photo_mask_test_1: + photo_mask_1: <<: *project_template - build: - context: . - container_name: photo_mask_test_1 - hostname: photo_mask_test_1 + container_name: photo_mask_1 + hostname: photo_mask_1 depends_on: - - ocr - - clas_orientation + - photo_review_1 command: [ 'photo_mask.py', '--clean', 'True' ] # photo_mask_2: