-
Notifications
You must be signed in to change notification settings - Fork 1
/
test.py
27 lines (23 loc) · 623 Bytes
/
test.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
import pickle
import numpy as np
import matplotlib.pyplot as plt
root_path = '/home/kun/Desktop/miccai21/TeCNO/Videos/dataframes/cholec_split_250px_25fps.pkl'
df = open(root_path,'rb')
data = pickle.load(df)
df.close()
print(data.keys())
print(data['video_idx'][:1000])
# pre = np.argmax(data[1], 1)
# gt = data[2]
#
# print(pre.shape[0])
# x = np.arange(pre.shape[0])
# l1 = plt.plot(x, pre, 'r--', label='pred')
# l2 = plt.plot(x, gt, 'b--', label='gt')
#
# plt.xlabel('row')
# plt.ylabel('column')
# plt.legend()
# # plt.savefig('./visual_phase/' + str(ID) + '.png')
# plt.show()
# exit()