fix autoextract

auto_extract.py

@@ -2,10 +2,14 @@ import os
 import sys
 import glob
 import time
+import threading
 
 
 hdd_root = '/hdd/*'
 
+def cmd_run(shell_cmd):
+    os.system(str(shell_cmd))
+
 def main():
     surgery_list = glob.glob(hdd_root)
 
@@ -16,19 +20,23 @@ def main():
             pass
 
     for id in surgery_list:
+        
         file_list=glob.glob(id + '/*')
         for file in file_list:
             if 'log.txt' in file:
                 f = open(file, 'rt')
                 logs = f.readlines()
-                f.close()
+                f.close()                
                 if len(logs) == 2:
+                    thread_list = []
                     f = open(file, 'at')
                     r_time = time.ctime(time.time())
                     f.write('exectue is successfully starts at ' + str(r_time) + '\n')
                     f.close()
-                    os.system('docker run --rm --name extract_1 --gpus '"device=0"' -v /dev:/dev -v /home/mc365/sources:/sources -v /hdd:/hdd ellishuntingmoon/mailsys:0.4 python3 /sources/extract_01.py ' + id + '/ >> /dev/null')
-                    os.system('docker run --rm --name extract_2 --gpus '"device=1"' -v /dev:/dev -v /home/mc365/sources:/sources -v /hdd:/hdd ellishuntingmoon/mailsys:0.4 python3 /sources/extract_02.py ' + id + '/ >> /dev/null')
+                    cmd_list = ['docker run --rm --name extract_1 --gpus '"device=0"' -v /dev:/dev -v /home/mc365/sources:/sources -v /hdd:/hdd ellishuntingmoon/mailsys:0.4 python3 /sources/extract_01.py ' + id + '/ >> /dev/null', 'docker run --rm --name extract_2 --gpus '"device=1"' -v /dev:/dev -v /home/mc365/sources:/sources -v /hdd:/hdd ellishuntingmoon/mailsys:0.4 python3 /sources/extract_02.py ' + id + '/ >> /dev/null']
+                    for index in range(0, len(cmd_list)):
+                        thread_list.append(threading.Thread(target=cmd_run,args=cmd_list))
+                        thread_list[index].start()                    
                     f = open(file, 'at')
                     r_time = time.ctime(time.time())
                     f.write('exectue is successfully ends at ' + str(r_time) + '\n')

extract_01.py

@@ -16,7 +16,7 @@ for svo in svo_list:
   if svo[-5:] == '1.svo':   
     os.system('python3 /sources/depthsensing.py -c /sources/cfg.cfg -w /sources/mailsys.weights -m /sources/data.data -s ' + target_dir + svo + ' -o ' + target_dir + svo[:5] + '.csv')
   else:
-    continue
+    pass
 
 full_list = os.listdir(target_dir)
 csv_1_list = [file for file in full_list if file.endswith("1.csv")]

extract_02.py

@@ -15,7 +15,7 @@ for svo in svo_list:
   if svo[-5:] == '2.svo':
     os.system('python3 /sources/depthsensing.py -c /sources/cfg.cfg -w /sources/mailsys.weights -m /sources/data.data -s ' + target_dir + svo + ' -o ' + target_dir + svo[:5] + '.csv')
   else:
-    continue
+    pass
 
 full_list = os.listdir(target_dir)
 csv_2_list = [file for file in full_list if file.endswith("2.csv")]