diff --git a/configs/SNEMI-Neuron-Train.yaml b/configs/SNEMI-Neuron-Train.yaml index ee4617ac..b6c5ef5b 100755 --- a/configs/SNEMI-Neuron-Train.yaml +++ b/configs/SNEMI-Neuron-Train.yaml @@ -13,9 +13,9 @@ MODEL: TARGET_OPT: ['2'] WEIGHT_OPT: [['1']] DATASET: - IMAGE_NAME: 'train.h5' - LABEL_NAME: 'label.h5' - INPUT_PATH: '/path/to/snemi/' + IMAGE_NAME: 'img/train-input.tif' + LABEL_NAME: 'label/train_labels.tif' + INPUT_PATH: '/n/pfister_lab2/Lab/vcg_connectomics/snemi/' OUTPUT_PATH: 'outputs/SNEMI/' PAD_SIZE: [4, 128, 128] AUGMENTOR: @@ -33,4 +33,4 @@ SOLVER: ITERATION_STEP: 1 ITERATION_SAVE: 1000 ITERATION_TOTAL: 40000 - SAMPLES_PER_BATCH: 8 \ No newline at end of file + SAMPLES_PER_BATCH: 8