Increased range of validation sections

atlas_server/src/static/training/config.py

@@ -11,8 +11,19 @@ def make_training_parameters(**kwargs):
     labels = ["bg", "cor", "wm", ] + __LABELS__
     sections = __SECTIONS__
     train_slices = [(brain, s) for s in sections]
-    validation_sections = get_slices_in_range(start=min(sections),
-            stop=max(sections),
+
+    # Slightly increase the number of sections to create predictions for
+    increase = 0.05
+    from_section = min(sections)
+    to_section = max(sections)
+    section_range = to_section - from_section
+    section_range = int(math.ceil((1 + increase) * section_range))
+    increase = int(math.ceil(section_range / 2))
+    min_section = max(from_section - increase, 1)
+    max_section = to_section + increase
+
+    validation_sections = get_slices_in_range(start=min_section,
+            stop=max_section,
             brain=brain,
             pattern=constants.original_path,
             step=1)
@@ -114,7 +125,7 @@ def make_training_parameters(**kwargs):
                         ),
                     ),
                 config.VAL_SPLIT: config.IteratorParams(
-                    batch_size=3,
+                    batch_size=12,
                     data_sources=[ds.ImageDataSource(name="gray2", size=8169, spacing=2, channels=CHANNEL_HISTOLOGY, data_format="NCHW"),
                         ds.ImageDataSource(name="gray16", size=1396, spacing=16, channels=CHANNEL_HISTOLOGY, data_format="NCHW"),
                         ds.LabelDataSource(name="labels", size=68, spacing=16, labels=labels, create=False)],