39 lines
934 B
Python
39 lines
934 B
Python
# 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
|