forked from ML4HPC/Brain_fMRI
/
readcsv_dti.py
44 lines (33 loc) · 1.37 KB
/
readcsv_dti.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import os
import csv
import numpy as np
import argparse
from readcsv import readcsv
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Read target csv for dti')
parser.add_argument('--data_dir', help='Path to dataset images')
parser.add_argument('--output_dir', help='Path to directory for saving outputs')
args = parser.parse_args()
train = "train_fluid_intelligence_household.csv"
valid = "valid_fluid_intelligence_household.csv"
test = "test_fluid_intelligence_household.csv"
if not os.path.isdir(args.output_dir):
try:
os.mkdir(args.output_dir)
except:
raise Exception('Could not create output directory')
csv_train = readcsv(args.data_dir, train)
csv_valid = readcsv(args.data_dir, valid)
csv_test = readcsv(args.data_dir, test)
for key, value in csv_train.items():
print(key, value)
for key, value in csv_valid.items():
print(key, value)
for key, value in csv_test.items():
print(key, value)
print('saving train dict!')
np.save(os.path.join(args.output_dir, 'csv_train_target_dti.npy'), csv_train)
print('saving valid dict!')
np.save(os.path.join(args.output_dir, 'csv_valid_target_dti.npy'), csv_valid)
print('saving test dict!')
np.save(os.path.join(args.output_dir, 'csv_test_target_dti.npy'), csv_test)