aydin/docker-compose.yml at main · royerlab/aydin · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# Docker Compose for Aydin
#
# Usage:
#   docker compose run --rm aydin denoise /data/image.tif       # CLI (CPU)
#   docker compose run --rm aydin-gpu denoise /data/image.tif   # CLI (GPU)
#   docker compose up aydin-studio                              # GUI at http://localhost:9876
#
# Place your images in ./data/ — they are mounted at /data inside the container.

services:

  # CLI — CPU-only (for classical denoisers, FGR, batch processing)
  aydin:
    platform: linux/amd64
    build:
      context: .
      target: aydin
    image: ghcr.io/royerlab/aydin:latest
    volumes:
      - ./data:/data

  # CLI — NVIDIA GPU (for CNN-based denoising)
  aydin-gpu:
    platform: linux/amd64
    build:
      context: .
      target: aydin-gpu
    image: ghcr.io/royerlab/aydin:gpu
    volumes:
      - ./data:/data
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

  # GUI — Aydin Studio accessible via web browser
  aydin-studio:
    platform: linux/amd64
    build:
      context: .
      target: aydin-studio
    image: ghcr.io/royerlab/aydin-studio:latest
    ports:
      - "9876:9876"
    volumes:
      - ./data:/data
    shm_size: '256m'
    environment:
      - XPRA_PORT=9876
      - XPRA_SCREEN=1920x1080x24+32