run_BSBT.py

import subprocess
import time
import numpy as np
from config import *

def run_BSBT(seq, rp, bSaveImage):
    x = seq.init_rect[0] - 1
    y = seq.init_rect[1] - 1
    w = seq.init_rect[2]
    h = seq.init_rect[3]

    path = './results/'

    if not os.path.exists(path):
        os.makedirs(path)

    command = list(map(str,['BeyondSemiBoostingTracker.exe', '100', '0.99', '2', 
        '0', '0', '0', seq.name, seq.path, seq.startFrame, seq.endFrame, 
        seq.nz, seq.ext, x, y, w, h]))

    tic = time.clock()
    subprocess.call(command)
    duration = time.clock() - tic

    result = dict()
    res = np.loadtxt('{0}_BSBT.txt'.format(seq.name), dtype=int)
    result['res'] = res.tolist()
    result['type'] = 'rect'
    result['fps'] = round(seq.len / duration, 3)

    return result