-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathplot_dataset.py
79 lines (71 loc) · 2.61 KB
/
plot_dataset.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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
# -*- coding: UTF-8 -*- #
"""
@filename:plot_dataset.py
@author:201300086
@time:2022-11-21
"""
from pedestrian_data import PedestrianDataset
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
# 步行轨迹图
CUT_BEGIN = 0 # 删掉前几秒的数据，因为GPS不准
def plot_locus(Lati,Longi,label="TBD",cut_begin=CUT_BEGIN):
Lati = Lati[cut_begin:]
Longi = Longi[cut_begin:]
# plt.xlabel("Latitude (°)")
# plt.ylabel("Longitude (°)")
plt.text(Lati[0], Longi[0], 's', fontsize=10)
plt.text(Lati[-1], Longi[-1], 'e', fontsize=10)
plt.plot(Lati, Longi,'+', markersize=1, label=label, )#
plt.legend(loc=0)
def plot_locus_realtime(Lati,Longi,label="TBD",cut_begin=CUT_BEGIN):
plt.clf() # 清除之前画的图
Lati = Lati[cut_begin:]
Longi = Longi[cut_begin:]
# plt.xlabel("Latitude (°)")
# plt.ylabel("Longitude (°)")
plt.text(Lati[0], Longi[0], 's', fontsize=10)
plt.text(Lati[-1], Longi[-1], 'e', fontsize=10)
plt.plot(Lati, Longi,'+', markersize=1, label=label, )#
plt.legend(loc=0)
plt.pause(0.001) # 暂停一段时间，不然画的太快会卡住显示不出来
plt.ioff() # 关闭画图窗口
# 重力加速度变化图
def plot_gravity(sample:dict,title="TBD"):
a = sample['Accelerometer']
b = sample['Linear Acceleration']
minus = (a - b)
print(minus)
c = np.array(list(map(lambda x: np.linalg.norm(x), minus)))
print(c.mean())
plt.plot(range(len(c)), c)
plt.title(title)
# l, r = (0, 9) # 画图范围
# dataset = PedestrianDataset(["Hand-Walk"], window_size=1000)
#
# for num, (name, locus) in enumerate(dataset):
# if num in range(l, r):
# print("正在遍历移动轨迹{}... \n".format(name))
# plt.subplot(33 * 10 + num%10 + 1)
# for sample in locus:
# #print(len(locus))
# plot_gravity(sample,title="{}".format(name))
# break
# if num >= r:
# break
# plt.show()
if __name__ == "__main__":
matplotlib.use('TkAgg')
l, r = (0, 9) # 画图范围
dataset = PedestrianDataset(["TestSet"], window_size=100,skip_len=5) # 指定文件夹
for num, (name, locus) in enumerate(dataset):
if num in range(l, r):
print("正在遍历移动轨迹{}... \n".format(name))
plt.subplot(33 * 10 + num % 10 + 1)
locus_pair = np.array(list(zip(locus.y_frame["Longitude (°)"], locus.y_frame["Latitude (°)"])))
print("轨迹长度: ", len(locus_pair))
plot_locus(locus_pair.T[0],locus_pair.T[1], label="{}".format(name))
if num >= r:
break
plt.show()