Ejemplo n.º 1
0
    def initOut(self):
        print '     Loading Data...'
        self.a_train_set = ADatasetFromFolder(
            sequence_dir, '../MOT/MOT16/test/MOT16-%02d' % self.seq_index)
        self.m_train_set = MDatasetFromFolder(
            sequence_dir, '../MOT/MOT16/test/MOT16-%02d' % self.seq_index)

        detection_dir = self.out_dir + 'res_training_det.txt'
        res_training = self.out_dir + 'res_training.txt'  # the result of the training data
        self.createTxt(detection_dir)
        self.createTxt(res_training)
        self.copyLines(self.seq_index, 1, detection_dir, self.tt, 1)

        self.evaluation(1, self.tt, detection_dir, res_training)
Ejemplo n.º 2
0
    def initOut(self):
        print('     Loading Data...')
        self.a_train_set = ADatasetFromFolder(sequence_dir, mot_dataset_dir + 'MOT16/test/MOT16-%02d' % self.seq_index,
                                              tau_conf_score)
        self.m_train_set = MDatasetFromFolder(sequence_dir, mot_dataset_dir + 'MOT16/test/MOT16-%02d' % self.seq_index,
                                              tau_conf_score)

        detection_dir = self.out_dir + 'res_det.txt'
        res_training = self.out_dir + 'res.txt'  # the tracking results
        self.createTxt(detection_dir)
        self.createTxt(res_training)
        self.copyLines(self.seq_index, 1, detection_dir, self.seq_len, 1)

        self.evaluation(1, self.seq_len, detection_dir, res_training)
Ejemplo n.º 3
0
    def initOut(self):
        for i in xrange(1, 9):
            start_t = time.time()
            self.camera = i
            self.bbx_counter = 0

            print '     Loading the model...'
            self.loadAModel()
            self.loadMModel()

            print '     Loading Data...'
            self.a_train_set = ADatasetFromFolder(i, self.tau_conf_score)
            self.m_train_set = MDatasetFromFolder(i, self.tau_conf_score)

            self.evaluation()
            print '     This camera:', self.camera, '- Time:', (time.time() - start_t)/60