def print_output(self):
for i, frame_name in enumerate(self.frames_number):
img = fn.get_frame(frame_name)
ca.CrowdArt(img, self.frames_p[i], 0, self.p_min, self.p_max, frame_name, 'Alem_P/pressure')
ca.CrowdArt(img, self.frames_v[i], 0, self.v_min, self.v_max, frame_name, 'Alem_V/velocity')
ca.CrowdArt(img, self.frames_d[i], 0, self.d_min, self.d_max, frame_name, 'Alem_D/density')
print '\r Frame ' + str(i + 1) + ' of ' + str(len(self.frames_number)) + ' saved',
def __init__(self, img):
self.crowded_image = fn.get_frame(img)
self.delta_x = int(self.crowded_image.shape[1] / mw.x_square)
self.delta_y = int(self.crowded_image.shape[0] / mw.y_square)
self.x_range = np.arange(self.delta_x, self.crowded_image.shape[1] + self.delta_x, self.delta_x)
self.y_range = np.arange(self.delta_y, self.crowded_image.shape[0] + self.delta_y, self.delta_y)
self.crowded_image = self.crowded_image[0: self.delta_y * mw.y_square, 0:self.delta_x * mw.x_square]
self.altitude = len(self.y_range)
self.count_matrix = np.zeros((len(self.x_range), len(self.y_range), 3), dtype=int)
self.total_SIFT = 0
self.total_FOUR = 0
self.xi = 0
self.yi = 0
