File size: 719 Bytes
5db43ff | 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 | import numpy as np
class TransSmooth:
def __init__(self):
self.trans_list = []
self.smooth_trans_list = []
def __len__(self):
return len(self.trans_list)
def append(self,trans):
self.trans_list.append(trans)
def offline_smooth(self):
length = len(self.trans_list)
radius = 6
for i in range(length):
count = 0
sum = np.zeros((2,3))
for j in range(i - radius, i + radius + 1):
if length > j >= 0:
sum += self.trans_list[j]
count += 1
self.smooth_trans_list.append(sum / count)
def get(self,i):
return self.smooth_trans_list[i]
|