map main paragraph to section paragraph

color_map.conf

@@ -48,8 +48,15 @@
 22=(220,20,60)
 #reference_list_title - #b8860b
 23=(184,134,11)
+
 #main_paragraph - #ff1493
-24=(255,20,147)
+#moved to alias
+#24=(255,20,147)
+
+[color_alias]
+# aliases are processed within the preprocessing step
+#main_paragraph - almost same as section_paragraph
+24=(128,0,128)
 
 [color_labels]
 1=abstract
@@ -75,5 +82,5 @@
 21=acknowledgement
 22=reference
 23=reference_list_title
-24=main_paragraph
+# 24=main_paragraph
 

color_map_selected.conf

@@ -48,8 +48,15 @@
 #22=(220,20,60)
 #reference_list_title - #b8860b
 #23=(184,134,11)
+
 #main_paragraph - #ff1493
-24=(255,20,147)
+#moved to alias
+#24=(255,20,147)
+
+[color_alias]
+# aliases are processed within the preprocessing step
+#main_paragraph - almost same as section_paragraph
+24=(128,0,128)
 
 [color_labels]
 1=abstract
@@ -75,5 +82,4 @@
 #21=acknowledgement
 #22=reference
 #23=reference_list_title
-24=main_paragraph
-
+#24=main_paragraph

sciencebeam_gym/trainer/preprocess.py

@@ -43,7 +43,9 @@ slim = tf.contrib.slim
 
 error_count = Metrics.counter('main', 'errorCount')
 
-def parse_color_map(f):
+def parse_color_map(f, section_names=None):
+  if section_names is None:
+    section_names = ['color_map']
   color_map_config = ConfigParser()
   color_map_config.readfp(f)
 
@@ -62,8 +64,10 @@ def parse_color_map(f):
     raise Exception('invalid color value: {}'.format(s))
 
   color_map = dict()
-  for k, v in color_map_config.items('color_map'):
-    color_map[parse_color(k)] = parse_color(v)
+  for section_name in section_names:
+    if color_map_config.has_section(section_name):
+      for k, v in color_map_config.items(section_name):
+        color_map[parse_color(k)] = parse_color(v)
   return color_map
 
 def map_colors(img, color_map):
@@ -198,7 +202,7 @@ def configure_pipeline(p, opt):
   color_map = None
   if opt.color_map:
     with file_io.FileIO(opt.color_map, 'r') as config_f:
-      color_map = parse_color_map(config_f)
+      color_map = parse_color_map(config_f, section_names=['color_alias', 'color_map'])
   if color_map:
     logger.info('read {} color mappings'.format(len(color_map)))
   else: