# image shape: this can be modified

VOLUME_ROWS = 256

VOLUME_COLS = 128

VOLUME_DEPS = 256

# number of classes

NUM_CLASSES = 4

# patch extract

PATCH_SIZE = 64

if PATCH_SIZE==64:

    EXTRACTTION_STEP = 12

    EXTRACTTION_STEP_CSF = 5

elif PATCH_SIZE==32:

    EXTRACTTION_STEP = 4

    EXTRACTTION_STEP_CSF = 4

elif PATCH_SIZE==16:

    EXTRACTTION_STEP = 9

    EXTRACTTION_STEP_CSF = 4

# training configs

UNET_MODEL = 0

if UNET_MODEL==0:

    MODEL = 'default'    

elif UNET_MODEL==1:

    MODEL = 'reduced'

elif UNET_MODEL==2:

    MODEL = 'extended'   

BASE = PATCH_SIZE

SMOOTH = 1.

NUM_EPOCHS  = 500

BATCH_SIZE  = 64

extraction_reconstruct_step = PATCH_SIZE

if PATCH_SIZE==64:

    PATIENCE = 10  

elif PATCH_SIZE==32:

    PATIENCE = 20  

elif PATCH_SIZE==16:

    PATIENCE = 10

# output

IMAGE_TYPE = '2d_whole_image'