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)