Ejemplo n.º 1
0
 def __init__(self, img_dir, tracker_type, dataset_config):
     self.img_dir = img_dir
     self.tracker_type = tracker_type
     self.frame_list = get_img_list(img_dir)
     self.frame_list.sort()
     dataname = img_dir.split('/')[-2]
     self.gts = get_ground_truthes(img_dir[:-4])
     if dataname in dataset_config.frames.keys():
         start_frame, end_frame = dataset_config.frames[dataname][0:2]
         if dataname != 'David':
             self.init_gt = self.gts[start_frame - 1]
         else:
             self.init_gt = self.gts[0]
         self.frame_list = self.frame_list[start_frame - 1:end_frame]
     else:
         self.init_gt = self.gts[0]
     if self.tracker_type == 'MOSSE':
         self.tracker = MOSSE()
     elif self.tracker_type == 'CSK':
         self.tracker = CSK()
     elif self.tracker_type == 'CN':
         self.tracker = CN()
     elif self.tracker_type == 'DSST':
         self.tracker = DSST()
     elif self.tracker_type == 'Staple':
         self.tracker = Staple(config=staple_config.StapleConfig())
     elif self.tracker_type == 'Staple-CA':
         self.tracker = Staple(config=staple_config.StapleCAConfig())
     elif self.tracker_type == 'KCF_CN':
         self.tracker = KCF(features='cn', kernel='gaussian')
     elif self.tracker_type == 'KCF_GRAY':
         self.tracker = KCF(features='gray', kernel='gaussian')
     elif self.tracker_type == 'KCF_HOG':
         self.tracker = KCF(features='hog', kernel='gaussian')
     elif self.tracker_type == 'DCF_GRAY':
         self.tracker = KCF(features='gray', kernel='linear')
     elif self.tracker_type == 'DCF_HOG':
         self.tracker = KCF(features='hog', kernel='linear')
     elif self.tracker_type == 'DAT':
         self.tracker = DAT()
     elif self.tracker_type == 'ECO-HC':
         self.tracker = ECO(config=otb_hc_config.OTBHCConfig())
     elif self.tracker_type == 'ECO':
         self.tracker = ECO(config=otb_deep_config.OTBDeepConfig())
     elif self.tracker_type == 'BACF':
         self.tracker = BACF()
     elif self.tracker_type == 'CSRDCF':
         self.tracker = CSRDCF()
     elif self.tracker_type == 'SAMF':
         self.tracker = SAMF()
     elif self.tracker_type == 'LDES':
         self.tracker = LDES(ldes_config.LDESDemoLinearConfig())
     elif self.tracker_type == 'DSST_LP':
         self.tracker = DSST_LP()
     else:
         raise NotImplementedError
Ejemplo n.º 2
0
    def __init__(self, img_dir, tracker_type, dataset_config):
        self.img_dir = img_dir
        self.tracker_type = tracker_type

        if self.tracker_type == 'MOSSE':
            self.tracker = MOSSE()
        elif self.tracker_type == 'CSK':
            self.tracker = CSK()
        elif self.tracker_type == 'CN':
            self.tracker = CN()
        elif self.tracker_type == 'DSST':
            self.tracker = DSST(dsst_config.DSSTConfig())
        elif self.tracker_type == 'Staple':
            self.tracker = Staple(config=staple_config.StapleConfig())
        elif self.tracker_type == 'Staple-CA':
            self.tracker = Staple(config=staple_config.StapleCAConfig())
        elif self.tracker_type == 'KCF_CN':
            self.tracker = KCF(features='cn', kernel='gaussian')
        elif self.tracker_type == 'KCF_GRAY':
            self.tracker = KCF(features='gray', kernel='gaussian')
        elif self.tracker_type == 'KCF_HOG':
            self.tracker = KCF(features='hog', kernel='gaussian')
        elif self.tracker_type == 'DCF_GRAY':
            self.tracker = KCF(features='gray', kernel='linear')
        elif self.tracker_type == 'DCF_HOG':
            self.tracker = KCF(features='hog', kernel='linear')
        elif self.tracker_type == 'DAT':
            self.tracker = DAT()
        elif self.tracker_type == 'ECO-HC':
            self.tracker = ECO(config=otb_hc_config.OTBHCConfig())
        elif self.tracker_type == 'ECO':
            self.tracker = ECO(config=otb_deep_config.OTBDeepConfig())
        elif self.tracker_type == 'BACF':
            self.tracker = BACF()
        elif self.tracker_type == 'CSRDCF':
            self.tracker = CSRDCF(config=csrdcf_config.CSRDCFConfig())
        elif self.tracker_type == 'CSRDCF-LP':
            self.tracker = CSRDCF(config=csrdcf_config.CSRDCFLPConfig())
        elif self.tracker_type == 'SAMF':
            self.tracker = SAMF()
        elif self.tracker_type == 'LDES':
            self.tracker = LDES(ldes_config.LDESDemoLinearConfig())
        elif self.tracker_type == 'DSST-LP':
            self.tracker = DSST(dsst_config.DSSTLPConfig())
        elif self.tracker_type == 'MKCFup':
            self.tracker = MKCFup(config=mkcf_up_config.MKCFupConfig())
        elif self.tracker_type == 'MKCFup-LP':
            self.tracker = MKCFup(config=mkcf_up_config.MKCFupLPConfig())
        elif self.tracker_type == 'STRCF':
            self.tracker = STRCF()
        elif self.tracker_type == 'MCCTH-Staple':
            self.tracker = MCCTHStaple(
                config=mccth_staple_config.MCCTHOTBConfig())
        else:
            raise NotImplementedError
Ejemplo n.º 3
0
def track_otb(tracker_type, dataset):
    for video in dataset:
        regions = []
        print('video:', video.name)
        if tracker_type == 'MOSSE':
            tracker = MOSSE()
        elif tracker_type == 'CSK':
            tracker = CSK()
        elif tracker_type == 'CN':
            tracker = CN()
        elif tracker_type == 'DSST':
            tracker = DSST()
        elif tracker_type == 'SAMF':
            tracker = SAMF()
        elif tracker_type == 'Staple':
            tracker = Staple(config=staple_config.StapleConfig())
        elif tracker_type == 'Staple-CA':
            tracker = Staple(config=staple_config.StapleCAConfig())
        elif tracker_type == 'KCF':
            tracker = KCF(features='hog', kernel='gaussian')
        elif tracker_type == 'DCF':
            tracker = KCF(features='hog', kernel='linear')
        elif tracker_type == 'DAT':
            tracker = DAT()
        elif tracker_type == 'ECO-HC':
            tracker = ECO(config=otb_hc_config.OTBHCConfig())
        elif tracker_type == 'ECO':
            tracker = ECO(config=otb_deep_config.OTBDeepConfig())
        elif tracker_type == 'BACF':
            tracker = BACF()
        elif tracker_type == 'CSRDCF':
            tracker = CSRDCF()
        elif tracker_type == 'CSRDCF-LP':
            tracker = CSRDCF_LP()
        elif tracker_type == 'OPENCV_KCF':
            tracker = OpenCVCFTracker(name='KCF')
        elif tracker_type == 'OPENCV_MOSSE':
            tracker = OpenCVCFTracker(name='MOSSE')
        elif tracker_type == 'OPENCV-CSRDCF':
            tracker = OpenCVCFTracker(name='CSRDCF')
        elif tracker_type == 'LDES':
            tracker = LDES(ldes_config.LDESOTBLinearConfig())
        elif tracker_type == 'LDES-NoBGD':
            tracker = LDES(ldes_config.LDESOTBNoBGDLinearConfig())
        elif tracker_type == 'DSST-LP':
            tracker = DSST_LP()
        elif tracker_type == 'MKCFup':
            tracker = MKCFup()
        elif tracker_type == 'MKCFup-LP':
            tracker = MKCFupLP()
        else:
            raise NotImplementedError
        for idx, (img, gt_bbox) in enumerate(video):
            if idx == 0:
                # init your tracker here
                tracker.init(img, tuple(gt_bbox))
                regions.append(gt_bbox)
                location = gt_bbox
            else:
                bbox = tracker.update(img)
                regions.append(bbox)
                location = bbox
            if args.visualization and idx >= 0:  # visualization (skip lost frame)
                im_show = img.copy()
                if idx == 0: cv2.destroyAllWindows()

                cv2.rectangle(
                    im_show, (gt_bbox[0], gt_bbox[1]),
                    (gt_bbox[0] + gt_bbox[2], gt_bbox[1] + gt_bbox[3]),
                    (0, 255, 0), 3)

                location = [int(l) for l in location]
                cv2.rectangle(
                    im_show, (location[0], location[1]),
                    (location[0] + location[2], location[1] + location[3]),
                    (0, 255, 255), 3)
                cv2.putText(im_show, str(idx), (40, 40),
                            cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 255), 2)

                cv2.imshow(video.name, im_show)
                cv2.waitKey(1)

        name = tracker_type

        video_path = join('test', args.dataset, name)
        if not isdir(video_path): makedirs(video_path)
        result_path = join(video_path, '{:s}.txt'.format(video.name))
        with open(result_path, "w") as fin:
            for x in regions:
                fin.write(','.join([str(i) for i in x]) + '\n')
Ejemplo n.º 4
0
    def __init__(self,img_dir,tracker_type,dataset_config):

        # img_directory is an attribute
        self.img_dir=img_dir

        # tracker type definition
        self.tracker_type=tracker_type

        # get_img_list function appends the name of all jpg files into a list.
        self.frame_list = get_img_list(img_dir)

        # Sorting the frame_list
        self.frame_list.sort()

        # name of the folder
        dataname=img_dir.split('/')[-2]

        # Getting the ground truth
        self.gts=get_ground_truthes(img_dir[:-4])


        if dataname in dataset_config.frames.keys():
            start_frame,end_frame=dataset_config.frames[dataname][0:2]
            if dataname!='David':
                self.init_gt=self.gts[start_frame-1]
            else:
                self.init_gt=self.gts[0]
            self.frame_list=self.frame_list[start_frame-1:end_frame]
        else:
            self.init_gt=self.gts[0]

        # Calling the appropriate tracker class as asked for.
        if self.tracker_type == 'MOSSE':
            self.tracker=MOSSE()
        elif self.tracker_type=='CSK':
            self.tracker=CSK()
        elif self.tracker_type=='CN':
            self.tracker=CN()
        elif self.tracker_type=='DSST':
            self.tracker=DSST(dsst_config.DSSTConfig())
        elif self.tracker_type=='Staple':
            self.tracker=Staple(config=staple_config.StapleConfig())
        elif self.tracker_type=='Staple-CA':
            self.tracker=Staple(config=staple_config.StapleCAConfig())
        elif self.tracker_type=='KCF_CN':
            self.tracker=KCF(features='cn',kernel='gaussian')
        elif self.tracker_type=='KCF_GRAY':
            self.tracker=KCF(features='gray',kernel='gaussian')
        elif self.tracker_type=='KCF_HOG':
            self.tracker=KCF(features='hog',kernel='gaussian')
        elif self.tracker_type=='DCF_GRAY':
            self.tracker=KCF(features='gray',kernel='linear')
        elif self.tracker_type=='DCF_HOG':
            self.tracker=KCF(features='hog',kernel='linear')
        elif self.tracker_type=='DAT':
            self.tracker=DAT()
        elif self.tracker_type=='ECO-HC':
            self.tracker=ECO(config=otb_hc_config.OTBHCConfig())
        elif self.tracker_type=='ECO':
            self.tracker=ECO(config=otb_deep_config.OTBDeepConfig())
        elif self.tracker_type=='BACF':
            self.tracker=BACF()
        elif self.tracker_type=='CSRDCF':
            self.tracker=CSRDCF(config=csrdcf_config.CSRDCFConfig())
        elif self.tracker_type=='CSRDCF-LP':
            self.tracker=CSRDCF(config=csrdcf_config.CSRDCFLPConfig())
        elif self.tracker_type=='SAMF':
            self.tracker=SAMF()
        elif self.tracker_type=='LDES':
            self.tracker=LDES(ldes_config.LDESDemoLinearConfig())
        elif self.tracker_type=='DSST-LP':
            self.tracker=DSST(dsst_config.DSSTLPConfig())
        elif self.tracker_type=='MKCFup':
            self.tracker=MKCFup(config=mkcf_up_config.MKCFupConfig())
        elif self.tracker_type=='MKCFup-LP':
            self.tracker=MKCFup(config=mkcf_up_config.MKCFupLPConfig())
        elif self.tracker_type=='STRCF':
            self.tracker=STRCF()
        elif self.tracker_type=='MCCTH-Staple':
            self.tracker=MCCTHStaple(config=mccth_staple_config.MCCTHOTBConfig())
        elif self.tracker_type=='MCCTH':
            self.tracker=MCCTH(config=mccth_config.MCCTHConfig())
        else:
            raise NotImplementedError