Paddle ocr, easyicr and doctr gpu support. (#4)

src/docker-compose.tuning.yml

@@ -0,0 +1,82 @@
+# docker-compose.tuning.yml - Ray Tune with all OCR services (PaddleOCR + DocTR)
+# Usage:
+#   docker compose -f docker-compose.tuning.yml up -d paddle-ocr-gpu doctr-gpu
+#   docker compose -f docker-compose.tuning.yml run raytune --service paddle --samples 64
+#   docker compose -f docker-compose.tuning.yml run raytune --service doctr --samples 64
+#   docker compose -f docker-compose.tuning.yml down
+#
+# Note: EasyOCR uses port 8002 (same as PaddleOCR). Use docker-compose.tuning.easyocr.yml separately.
+
+services:
+  raytune:
+    image: seryus.ddns.net/unir/raytune:latest
+    network_mode: host
+    shm_size: '5gb'
+    volumes:
+      - ./results:/app/results:rw
+    environment:
+      - PYTHONUNBUFFERED=1
+
+  paddle-ocr-gpu:
+    image: seryus.ddns.net/unir/paddle-ocr-gpu:latest
+    container_name: paddle-ocr-gpu-tuning
+    ports:
+      - "8002:8000"
+    volumes:
+      - ./dataset:/app/dataset:ro
+      - ./debugset:/app/debugset:rw
+      - paddlex-cache:/root/.paddlex
+    environment:
+      - PYTHONUNBUFFERED=1
+      - CUDA_VISIBLE_DEVICES=0
+      - PADDLE_DET_MODEL=PP-OCRv5_mobile_det
+      - PADDLE_REC_MODEL=PP-OCRv5_mobile_rec
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
+    restart: unless-stopped
+    healthcheck:
+      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8000/health')"]
+      interval: 30s
+      timeout: 10s
+      retries: 3
+      start_period: 60s
+
+  doctr-gpu:
+    image: seryus.ddns.net/unir/doctr-gpu:latest
+    container_name: doctr-gpu-tuning
+    ports:
+      - "8003:8000"
+    volumes:
+      - ./dataset:/app/dataset:ro
+      - ./debugset:/app/debugset:rw
+      - doctr-cache:/root/.cache/doctr
+    environment:
+      - PYTHONUNBUFFERED=1
+      - CUDA_VISIBLE_DEVICES=0
+      - DOCTR_DET_ARCH=db_resnet50
+      - DOCTR_RECO_ARCH=crnn_vgg16_bn
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
+    restart: unless-stopped
+    healthcheck:
+      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8000/health')"]
+      interval: 30s
+      timeout: 10s
+      retries: 3
+      start_period: 180s
+
+volumes:
+  paddlex-cache:
+    name: paddlex-model-cache
+  doctr-cache:
+    name: doctr-model-cache