-
Notifications
You must be signed in to change notification settings - Fork 1
/
tester.py
39 lines (23 loc) · 801 Bytes
/
tester.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
# %%
import pickle
file = 'deepspt_results/analytics/frame_change_pruned_DeepSPT.pkl'
f = pickle.load(open(file, 'rb'))
f1 = [t for t in f]
pickle.dump(f1,
open('deepspt_results/analytics/frame_change_pruned_DeepSPT_v1.pkl', 'wb'))
file = 'deepspt_results/analytics/AP2_tracks_DeepSPT.pkl'
f = pickle.load(open(file, 'rb'))
f1 = [t for t in f]
pickle.dump(f1,
open('deepspt_results/analytics/AP2_tracks_DeepSPT_v1.pkl', 'wb'))
# %%
# load npy file
import numpy as np
path = '/Users/jacobkh/Documents/PhD/SPT/github_final/DeepSPT/_For_publication/baseline_methods/'
file_GT = 'hmm_rota_changepoints_GT.npy'
file_ML = 'hmm_rota_changepoints_ML.npy'
GT = np.load(path+file_GT, allow_pickle=True)
ML = np.load(path+file_ML, allow_pickle=True)
GT, ML
# %%
12.61**2