rename train_infer as net_run

Author: taigw

pymic/train_infer/__init__.py pymic/net_run/__init__.pypymic/train_infer/__init__.py renamed to pymic/net_run/__init__.py

@@ -18,10 +18,10 @@
 from pymic.io.image_read_write import save_nd_array_as_image
 from pymic.io.nifty_dataset import NiftyDataset
 from pymic.io.transform3d import get_transform
-from pymic.train_infer.net_factory import get_network
-from pymic.train_infer.infer_func import volume_infer
-from pymic.train_infer.loss import *
-from pymic.train_infer.get_optimizer import get_optimiser
+from pymic.net_run.net_factory import get_network
+from pymic.net_run.infer_func import volume_infer
+from pymic.net_run.loss import *
+from pymic.net_run.get_optimizer import get_optimiser
 from pymic.util.image_process import convert_label
 from pymic.util.parse_config import parse_config
 
@@ -387,7 +387,7 @@ def run(self):
         else:
             self.infer()
 
-if __name__ == "__main__":
+def main():
     if(len(sys.argv) < 3):
         print('Number of arguments should be 3. e.g.')
         print('    python train_infer.py train config.cfg')
@@ -398,3 +398,7 @@ def run(self):
     agent    = TrainInferAgent(config, stage)
     agent.run()
 
+if __name__ == "__main__":
+    main()
+    
+

pymic/train_infer/get_optimizer.py pymic/net_run/get_optimizer.pypymic/train_infer/get_optimizer.py renamed to pymic/net_run/get_optimizer.py

@@ -324,12 +324,16 @@ def evaluation(config_file):
         np.savetxt("{0:}/{1:}_{2:}_std.txt".format(s_folder, organ_name, metric), score_std)
         print("{0:} mean ".format(metric), score_mean)
         print("{0:} std  ".format(metric), score_std) 
-    
-if __name__ == '__main__':
+
+
+def main():
     if(len(sys.argv) < 2):
         print('Number of arguments should be 2. e.g.')
         print('    python pyMIC.util/evaluation.py config.cfg')
         exit()
     config_file = str(sys.argv[1])
     assert(os.path.isfile(config_file))
     evaluation(config_file)
+    
+if __name__ == '__main__':
+    main()

pymic/train_infer/infer_func.py pymic/net_run/infer_func.pypymic/train_infer/infer_func.py renamed to pymic/net_run/infer_func.py

@@ -27,4 +27,10 @@
         'Programming Language :: Python :: 3',
     ],
     python_requires = '>=3.6',
+    entry_points = {
+        'console_scripts': [
+            'pymic_net_run  = pymic.net_run.net_run:main',
+            'pymic_evaluate = pymic.util.evaluation:main'
+        ],
+    },
 )