def gt_test(self): path = "D:\\workspace_pycharm\\20160202_mlv_mpp" path_dataset = path + '\\data\\CAVIAR' fname = 'WalkByShop1cor.mpg' raw_frame_list, video_props = read_video_as_list_by_path(path_dataset, fname) gt_list = self.gt_list assert (len(gt_list) == len(raw_frame_list)), 'length are different' print video_props display = [] for i in range(len(gt_list)): sublist = gt_list[i] for gt in sublist: x = gt[1] y = gt[2] w = gt[3] h = gt[4] cv2.rectangle(raw_frame_list[i], (x, y), (x + w, y + h), (200, 24, 100), 1) display.append(raw_frame_list[i]) display_img(display)
def run(self): self.raw_frame_list, self.video_props = read_video_as_list_by_path(self.path_dataset, self.cvaiar_list[0]) self.fg_list = get_list_fgmask(self.raw_frame_list, self.history, self.n_mixture, self.p_background) # self.run_extrinsics() self.run_intrinsics()