update train.py and predict.py

Confirm that the configuration file exists

Author: taigw

pymic/net_run/agent_cls.py

@@ -327,8 +327,8 @@ def infer(self):
 
         if(self.config['testing'].get('evaluation_mode', True)):
             self.net.eval()
-            
-        output_csv   = self.config['testing']['output_csv']
+        
+        output_csv   = self.config['testing']['output_dir'] + '/' + self.config['testing']['output_csv']
         class_num    = self.config['network']['class_num']
         save_probability = self.config['testing'].get('save_probability', False)
 

pymic/net_run/predict.py

@@ -17,6 +17,8 @@ def main():
         print('   pymic_test config.cfg')
         exit()
     cfg_file = str(sys.argv[1])
+    if(not os.path.isfile(cfg_file)):
+        raise ValueError("The config file does not exist: " + cfg_file)
     config   = parse_config(cfg_file)
     config   = synchronize_config(config)
     log_dir  = config['testing']['output_dir']

pymic/net_run/train.py

@@ -68,6 +68,8 @@ def main():
         print('   pymic_train config.cfg')
         exit()
     cfg_file = str(sys.argv[1])
+    if(not os.path.isfile(cfg_file)):
+        raise ValueError("The config file does not exist: " + cfg_file)
     config   = parse_config(cfg_file)
     config   = synchronize_config(config)
     log_dir  = config['training']['ckpt_save_dir']