def test_siamfc_track_v2(self):
dataset = VOT(self.vot_dir, return_rect=True, download=True)
tracker = TrackerSiamFC(
branch='alexv2', net_path=self.net_v2, z_lr=0.01,
response_up=8, scale_step=1.0816, window_influence=0.25)
img_files, anno = random.choice(dataset)
rects, speed = tracker.track(img_files, anno[0, :],
visualize=True)
self.assertEqual(rects.shape, anno.shape)
def test_siamfc_train_v2(self):
tracker = TrackerSiamFC(branch='alexv2')
transform = TransformSiamFC(
stats_path=self.stats_path, score_sz=33,
r_pos=8, total_stride=4)
base_dataset = VOT(self.vot_dir, return_rect=True, download=True)
dataset = Pairwise(base_dataset, transform, pairs_per_video=1)
dataloader = DataLoader(dataset, batch_size=2, shuffle=True)
# training loop
for it, batch in enumerate(dataloader):
update_lr = it == 0
loss = tracker.step(batch, backward=True, update_lr=update_lr)
print('Iter: {} Loss: {:.6f}'.format(it + 1, loss))
# val loop
for it, batch in enumerate(dataloader):
loss = tracker.step(batch, backward=False)
print('Val. Iter: {} Loss: {:.6f}'.format(it + 1, loss))
from __future__ import absolute_import import argparse from lib.trackers import TrackerSiamFC from lib.experiments import ExperimentOTB otb_dir = 'data/OTB' experiment = ExperimentOTB(otb_dir) net_path = 'pretrained/siamfc/2016-08-17.net.mat' tracker = TrackerSiamFC(branch='alexv1', net_path=net_path) experiment.run(tracker, visualize=True) experiment.report(tracker.name)
from __future__ import absolute_import
import json
from lib.trackers import TrackerSiamFC
from lib.experiments import ExperimentOTB
otb_dir = 'data/OTB'
experiment = ExperimentOTB(otb_dir, version=2013)
branch = 'alexv2'
config_file = 'config/siamfc.json'
with open(config_file, 'r') as f:
config = json.load(f)[branch]
net_path = 'pretrained/siamfc/baseline-conv5_e55.mat'
tracker = TrackerSiamFC(branch=branch, net_path=net_path, **config)
experiment.run(tracker, visualize=False)
experiment.report(['SiamFC'])
def test_experiment_unsupervised(self):
experiment = ExperimentUnsupervised(log_dir='results/SiamFC')
tracker = TrackerSiamFC(net_path=self.net_path)
dataset = VOT(self.vot_dir, return_rect=True)
experiment.run(tracker, dataset, visualize=True)