wound_segmentation.constants module

Global constants used across the segmentation project.

This module defines all key constants for consistent usage throughout the codebase, such as image size, default filenames, output paths, and download identifiers.

Variables

IMG_SIZEint

Input image size used for both model training and inference (square dimension).

DEFAULT_MODEL_FILENAMEstr

Default filename for the segmentation model weights.

DEFAULT_MODEL_DIRstr

Directory where model weights are stored or downloaded to.

DEFAULT_MODEL_PATHstr

Full path to the model weights file (directory + filename).

DEFAULT_OUTPUT_DIRstr

Directory where prediction results and overlay images are saved.

GDRIVE_FILE_IDstr

Google Drive ID used to download zipped model weights if missing locally.

Usage

These constants are imported by different modules to ensure unified configuration and path management.

Examples

>>> from constants import IMG_SIZE, DEFAULT_MODEL_PATH
>>> print(DEFAULT_MODEL_PATH)
'model_weights/segmentation_model.weights.h5'