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)
Beispiel #2
0
 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()