[ccb1dd]: / validate_data_order.py

Download this file

16 lines (11 with data), 513 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
import argparse
from fetal_net.data import open_data_file
parser = argparse.ArgumentParser()
parser.add_argument("--data1_path", help="specifies model path",
type=str, required=True)
parser.add_argument("--data2_path", help="specifies model path",
type=str, required=True)
opts = parser.parse_args()
ids_1 = open_data_file(opts.data1_path).root.subject_ids
ids_2 = open_data_file(opts.data2_path).root.subject_ids
print(all([i1 == i2 for i1, i2 in zip(ids_1, ids_2)]))