# DeepLabV3+ Configuration
import os

# 기본 경로 설정
BASE_DIR = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
DATA_DIR = os.path.join(BASE_DIR, 'data')

# 이미지 경로 (환경에 맞게 수정)
IMAGE_PATH = os.environ.get('DEEPLABV3_IMAGE_PATH', '/water_body_data/image/')
MASK_PATH = os.environ.get('DEEPLABV3_MASK_PATH', '/water_body_data/mask/')

# TIF 데이터셋 경로
TIF_DATASET_DIR = os.path.join(BASE_DIR, 'app', 'static', 'water_segmentation')

# 입력 채널 수 (R, G, B, MNDWI)
IN_CHANNELS = 4

# 모델 하이퍼파라미터
IMAGE_SIZE = 128
BATCH_SIZE = 8
LEARNING_RATE = 1e-3

# 학습 설정
EPOCHS = 50
SHUFFLE_BUFFER = 1000

# 모델 저장 경로
MODEL_SAVE_PATH = os.path.join(BASE_DIR, 'models', 'DeepLabV3-Plus.h5')

# ResNet50 설정
RESNET_WEIGHTS = 'imagenet'

# ASPP 설정
ASPP_FILTERS = 256
ASPP_DILATION_RATES = [1, 6, 12, 18]

# LLF 설정
LLF_FILTERS = 48