docker-compose.yml

version: "3.9"
services:
  rtsp_server:
    image: bluenviron/mediamtx:1.5.1-ffmpeg
    container_name: rtsp_server
    ports:
      - 8554:8554
      - 1935:1935
      - 8888:8888
      - 8889:8889
      - 8890:8890/udp
      - 8189:8189/udp
    volumes:
      - ./${STOCK_VIDEO_RELATIVE_PATH}:/stock_video.mp4
      - ./rtsp-server/mediamtx.yml:/mediamtx.yml

  flask_server:
    build:
      context: flask-app/
      dockerfile: Dockerfile
      args:
        - yolo_model=yolov5s.onnx
    container_name: flask_server
    ports:
      - 8080:8080
    volumes:
      - ./output:/output
    environment:
      - RTSP_URL=rtsp://rtsp_server:8554/stream
      - THREAD_RETRY_INTERVAL=5
      - IOU_THRESHOLD=0.5
      - OBJECT_DETECTION_SCORE=0.4
      - ONNX_FALLBACK_DOWNLOAD_URL=https://github.com/ultralytics/yolov5/releases/download/v7.0/yolov5s.onnx
      - ONNX_LOCAL_FILE_PATH=yolov5.onnx
    depends_on:
      - rtsp_server