Jeong Geol Kim 4 лет назад
Родитель
Сommit
d663642f1e
2 измененных файлов: 14 добавлений и 6 удалений
  1. 8
    3
      extract_01.py
  2. 6
    3
      extract_02.py

+ 8
- 3
extract_01.py Просмотреть файл

4
 import pandas as pd
4
 import pandas as pd
5
 
5
 
6
 #ssd_dir = /data/36536536520201231/
6
 #ssd_dir = /data/36536536520201231/
7
-ssd_dir = sys.argv[1]
7
+ssd_dir = str(sys.argv[1])
8
 try:
8
 try:
9
   os.system('mv '+ssd_dir+ ' /hdd')
9
   os.system('mv '+ssd_dir+ ' /hdd')
10
 except:
10
 except:
11
   pass
11
   pass
12
+
12
 target_dir = '/hdd' + ssd_dir[5:]
13
 target_dir = '/hdd' + ssd_dir[5:]
13
 
14
 
14
 
15
 
15
 svo_list = os.listdir(target_dir)
16
 svo_list = os.listdir(target_dir)
17
+svo_list = sorted(svo_list)
18
+
19
+
16
 
20
 
17
 for svo in svo_list:
21
 for svo in svo_list:
18
-  if svo[-5] == 1:
22
+  if svo[-5] == '1':
19
     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[:4] + '.csv')
23
     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[:4] + '.csv')
20
   else:
24
   else:
21
-    pass
25
+    continue
22
 
26
 
23
 full_list = os.listdir(target_dir)
27
 full_list = os.listdir(target_dir)
24
 csv_1_list = [file for file in full_list if file.endswith("1.csv")]
28
 csv_1_list = [file for file in full_list if file.endswith("1.csv")]
29
+csv_1_list = sorted(csv_1_list)
25
 
30
 
26
 conc_df = pd.DataFrame(data={'x': [], 'y': [], 'z': [], 'frame': []})
31
 conc_df = pd.DataFrame(data={'x': [], 'y': [], 'z': [], 'frame': []})
27
 FRAME_COUNTER = 0
32
 FRAME_COUNTER = 0

+ 6
- 3
extract_02.py Просмотреть файл

14
 
14
 
15
 
15
 
16
 svo_list = os.listdir(target_dir)
16
 svo_list = os.listdir(target_dir)
17
+svo_list = sorted(svo_list)
18
+
17
 
19
 
18
 for svo in svo_list:
20
 for svo in svo_list:
19
-  if svo[-5] == 2:
21
+  if svo[-5] == '2':
20
     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[:4] + '.csv')
22
     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[:4] + '.csv')
21
   else:
23
   else:
22
     pass
24
     pass
23
 
25
 
24
 full_list = os.listdir(target_dir)
26
 full_list = os.listdir(target_dir)
25
-csv_1_list = [file for file in full_list if file.endswith("1.csv")]
27
+csv_2_list = [file for file in full_list if file.endswith("2.csv")]
28
+csv_2_list = sorted(csv_2_list)
26
 
29
 
27
 conc_df = pd.DataFrame(data={'x': [], 'y': [], 'z': [], 'frame': []})
30
 conc_df = pd.DataFrame(data={'x': [], 'y': [], 'z': [], 'frame': []})
28
 FRAME_COUNTER = 0
31
 FRAME_COUNTER = 0
29
 N_OF_FRAME = 1800
32
 N_OF_FRAME = 1800
30
-for x in csv_1_list:
33
+for x in csv_2_list:
31
   df = pd.read_csv(os.path.join(target_dir,x), header=0)
34
   df = pd.read_csv(os.path.join(target_dir,x), header=0)
32
   df['frame'] = df['frame'] + FRAME_COUNTER
35
   df['frame'] = df['frame'] + FRAME_COUNTER
33
   conc_df = pd.concat([conc_df,df])
36
   conc_df = pd.concat([conc_df,df])

Загрузка…
Отмена
Сохранить