Fixed constant names in config.py

atlas_server/src/static/training/config.py

@@ -3,7 +3,7 @@ def make_training_parameters(**kwargs):
     import math
     from atlas.configuration import config, constants
     from atlas.configuration import data_source as ds
-    from atlas.data.image_provider import CHANNEL_GRAY
+    from atlas.data.image_provider import CHANNEL_HISTOLOGY
     from atlas.experiments import get_slices_in_range
 
     brain = __BRAIN__
@@ -75,8 +75,8 @@ def make_training_parameters(**kwargs):
         iterator_params={
             config.TRAIN_SPLIT: config.IteratorParams(
                 batch_size=16,
-                data_sources=[ds.ImageDataSource(name="gray2", size=2025, spacing=2, channels=CHANNEL_GRAY),
-                              ds.ImageDataSource(name="gray16", size=628, spacing=16, channels=CHANNEL_GRAY),
+                data_sources=[ds.ImageDataSource(name="gray2", size=2025, spacing=2, channels=CHANNEL_HISTOLOGY),
+                              ds.ImageDataSource(name="gray16", size=628, spacing=16, channels=CHANNEL_HISTOLOGY),
                               ds.LabelDataSource(name="labels", size=68, spacing=16, labels=labels)],
                 multi_threaded_communications=11,
                 image_provider_params=config.ImageProviderParams(
@@ -113,8 +113,8 @@ def make_training_parameters(**kwargs):
             ),
             config.VAL_SPLIT: config.IteratorParams(
                 batch_size=5,
-                data_sources=[ds.ImageDataSource(name="gray2", size=8169, spacing=2, channels=CHANNEL_GRAY),
-                              ds.ImageDataSource(name="gray16", size=1396, spacing=16, channels=CHANNEL_GRAY), ],
+                data_sources=[ds.ImageDataSource(name="gray2", size=8169, spacing=2, channels=CHANNEL_HISTOLOGY),
+                              ds.ImageDataSource(name="gray16", size=1396, spacing=16, channels=CHANNEL_HISTOLOGY), ],
                 multi_threaded_communications=11,
                 image_provider_params=config.ImageProviderParams(
                     transformation_parameters=config.TransformationParams(

atlas_server/src/static/training/train.job.sh

@@ -46,4 +46,4 @@ atlas img ann2hdf5 ${ATLAS_ANNOTATION_DIR}/*.json \
 atlas exp run new `pwd`
 
 # Train network
-srun atlas dl train `pwd` --verbose --distribute
\ No newline at end of file
+srun atlas dl train `pwd` --verbose --distribute