调整model的存放位置

This commit is contained in:
2024-07-18 16:23:58 +08:00
parent 9bd7038d5f
commit cb77f229ab
5 changed files with 148 additions and 6 deletions

51
docker-compose-mask.yml Normal file
View File

@@ -0,0 +1,51 @@
services:
photo_mask_1:
container_name: photo_mask_1
image: fcb_photo_review
volumes:
- ./log:/app/log
restart: always
depends_on:
- photo_review_1
command: [ "photo_mask.py", "--clean", "True" ]
deploy:
resources:
reservations:
devices:
- device_ids: [ "1" ]
capabilities: [ "gpu" ]
driver: "nvidia"
photo_mask_2:
container_name: photo_mask_2
image: fcb_photo_review
volumes:
- ./log:/app/log
restart: always
depends_on:
- photo_mask_1
command: [ "photo_mask.py" ]
deploy:
resources:
reservations:
devices:
- device_ids: [ "1" ]
capabilities: [ "gpu" ]
driver: "nvidia"
photo_mask_3:
container_name: photo_mask_3
image: fcb_photo_review
volumes:
- ./log:/app/log
restart: always
depends_on:
- photo_mask_2
command: [ "photo_mask.py" ]
deploy:
resources:
reservations:
devices:
- device_ids: [ "1" ]
capabilities: [ "gpu" ]
driver: "nvidia"