fuse net_run_cls and net_run_seg

Author: taigw

pymic/net_run/net_run_cls.py renamed to pymic/net_run/net_run.py

@@ -3,16 +3,22 @@
 import sys
 from pymic.util.parse_config import parse_config
 from pymic.net_run.agent_cls import ClassificationAgent
+from pymic.net_run.agent_seg import SegmentationAgent
 
 def main():
     if(len(sys.argv) < 3):
         print('Number of arguments should be 3. e.g.')
-        print('   pymic_net_run_cls train config.cfg')
+        print('   pymic_net_run train config.cfg')
         exit()
     stage    = str(sys.argv[1])
     cfg_file = str(sys.argv[2])
     config   = parse_config(cfg_file)
-    agent    = ClassificationAgent(config, stage)
+    task     = config['dataset']['task_type']
+    assert task in ['cls', 'cls_nexcl', 'seg']
+    if(task == 'cls' or task == 'cls_nexcl'):
+        agent = ClassificationAgent(config, stage)
+    else:
+        agent = SegmentationAgent(config, stage)
     agent.run()
 
 if __name__ == "__main__":