jeonghyo.k
3946ff6a25
- prediction/image/ FastAPI 서버 Docker 환경 구성 - Dockerfile: PyTorch 2.1 + CUDA 12.1 기반 GPU 이미지 - docker-compose.yml: GPU 할당 + 데이터 볼륨 마운트 - requirements.txt: 서버 의존성 목록 - .env.example: 환경변수 템플릿 - DOCKER_USAGE.md: 빌드/실행/API 사용법 문서 - Dockerfile에 .dockerignore 제외 폴더 mkdir -p 추가 - .gitignore: prediction/image 결과물 및 모델 가중치(.pth) 제외 추가 - dbInsert_csv.py, dbInsert_shp.py 삭제 (미사용 DB 로직) - api.py: dbInsert import 및 주석 처리된 DB 호출 코드 제거 - aerialRouter.ts: req.params 타입 오류 수정
49 lines
2.2 KiB
YAML
49 lines
2.2 KiB
YAML
version: "3.9"
|
|
|
|
services:
|
|
image-analysis:
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile
|
|
image: wing-image-analysis:latest
|
|
container_name: wing-image-analysis
|
|
ports:
|
|
- "5001:5001"
|
|
env_file: .env
|
|
|
|
volumes:
|
|
# ── mx15hdi (EO 드론 카메라) ────────────────────────────────────────
|
|
# 입력: 업로드된 원본 이미지
|
|
- ./mx15hdi/Metadata/Image/Original_Images:/app/mx15hdi/Metadata/Image/Original_Images
|
|
# 출력: 메타데이터 CSV
|
|
- ./mx15hdi/Metadata/CSV:/app/mx15hdi/Metadata/CSV
|
|
# 출력: 지리참조 GeoTIFF (컬러 / 마스크)
|
|
- ./mx15hdi/Georeference/Tif:/app/mx15hdi/Georeference/Tif
|
|
- ./mx15hdi/Georeference/Mask_Tif:/app/mx15hdi/Georeference/Mask_Tif
|
|
# 출력: 유류 폴리곤 Shapefile
|
|
- ./mx15hdi/Polygon/Shp:/app/mx15hdi/Polygon/Shp
|
|
# 출력: 블렌딩 추론 결과 / 마스크 이미지
|
|
- ./mx15hdi/Detect/result:/app/mx15hdi/Detect/result
|
|
- ./mx15hdi/Detect/Mask_result:/app/mx15hdi/Detect/Mask_result
|
|
# ── starsafire (열화상 카메라) ──────────────────────────────────────
|
|
- ./starsafire/Metadata/Image/Original_Images:/app/starsafire/Metadata/Image/Original_Images
|
|
- ./starsafire/Metadata/CSV:/app/starsafire/Metadata/CSV
|
|
- ./starsafire/Georeference/Tif:/app/starsafire/Georeference/Tif
|
|
- ./starsafire/Georeference/Mask_Tif:/app/starsafire/Georeference/Mask_Tif
|
|
- ./starsafire/Polygon/Shp:/app/starsafire/Polygon/Shp
|
|
- ./starsafire/Detect/result:/app/starsafire/Detect/result
|
|
- ./starsafire/Detect/Mask_result:/app/starsafire/Detect/Mask_result
|
|
# ── 스티칭 결과 ─────────────────────────────────────────────────────
|
|
- ./stitch:/app/stitch
|
|
|
|
# NVIDIA GPU 할당 (nvidia-container-toolkit 필수)
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: 1
|
|
capabilities: [gpu]
|
|
|
|
restart: unless-stopped
|