Initial commit

Flexbrdf/scripts/no_ray/run_single_process_trait.py

@@ -0,0 +1,45 @@
+
+import sys, os 
+import multiprocessing
+import subprocess, json
+
+exec_str="python ../no_ray/trait_estimate_inde.py "
+
+def run_command(command):
+    print(command)
+    subprocess.run(command,shell=True) 
+
+def main():
+
+    config_file = sys.argv[1]
+
+    total_img_count = int(sys.argv[2])
+    total_trait_count = int(sys.argv[3])
+
+    total_count = total_img_count*total_trait_count
+    worker_count = min(os.cpu_count()-1,total_count)
+
+    with open(config_file, 'r') as outfile:
+        config_dict = json.load(outfile)
+
+        pool = multiprocessing.Pool(processes=worker_count)
+
+        if (total_img_count > len(config_dict["input_files"])) or (total_trait_count> len(config_dict["input_files"])):
+            print("Out of upper bound")
+            return
+
+        param_list = []
+        for img_i in range(total_img_count):
+            for trait_j in range(total_trait_count):
+                param_list+=[[img_i,trait_j]]
+
+        commands=[f"{exec_str} {config_file} {param_order[0]} {param_order[1]}" for param_order in param_list]
+        pool.map(run_command, commands)
+        pool.close()
+        pool.join()  # Wait for all subprocesses to finish
+
+        print('All traits are done.')
+
+
+if __name__== "__main__":
+    main()