atnikos commited on
Commit
f71c08d
·
1 Parent(s): 38c4910

first tmr retrieval efffort

Browse files
Files changed (2) hide show
  1. dataset_utils.py +22 -0
  2. style.css +5 -0
dataset_utils.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from gen_utils import read_json
2
+
3
+ def load_motionfix(path_to_data):
4
+ dataset = read_json(path_to_data + '/amt_motionfix_latest.json')
5
+ splits = read_json(path_to_data + '/splits.json')
6
+
7
+ train_data = {}
8
+ val_data = {}
9
+ test_data = {}
10
+
11
+ # Fill each dictionary with the corresponding data
12
+ for key in splits['train']:
13
+ train_data[key] = dataset[key]
14
+
15
+ for key in splits['val']:
16
+ val_data[key] = dataset[key]
17
+
18
+ for key in splits['test']:
19
+ test_data[key] = dataset[key]
20
+ validation_test_data = {**val_data, **test_data}
21
+
22
+ return train_data, validation_test_data
style.css ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ /* style.css */
2
+ #center-column {
3
+ display: flex;
4
+ justify-content: center;
5
+ }