def __init__(self): super(SiamFCPP, self).__init__("SiamFC++") root_cfg.merge_from_file(path_config.SIAMFCPP_CONFIG) task = "track" task_cfg = root_cfg["test"][task] task_cfg.freeze() # build model model = model_builder.build(task, task_cfg.model) # build pipeline self.pipeline = pipeline_builder.build(task, task_cfg.pipeline, model) dev = torch.device("cuda") self.pipeline.set_device(dev)
def scan_key(): logger.info("Key usage prompt: press ESC for exit") key = cv2.waitKey(0) & 0xFF if key == 27: logger.info("ESC pressed, debugging terminated.") exit(0) if __name__ == '__main__': # parsing parser = make_parser() parsed_args = parser.parse_args() # experiment config exp_cfg_path = osp.realpath(parsed_args.config) root_cfg.merge_from_file(exp_cfg_path) logger.info("Load experiment configuration at: %s" % exp_cfg_path) logger.info( "Merged with root_cfg imported from videoanalyst.config.config.cfg") # resolve config root_cfg = root_cfg.train task, task_cfg = specify_task(root_cfg) task_cfg.data.num_workers = 2 task_cfg.data.sampler.submodules.dataset.GOT10kDataset.check_integrity = False task_cfg.freeze() if parsed_args.target == "dataloader": logger.info("visualize for dataloader") with Timer(name="Dataloader building", verbose=True): dataloader = dataloader_builder.build(task, task_cfg.data)
# multi Thread from numba import config, njit, threading_layer, jit import numba from util import hyper_params, postprocess_box, postprocess_score, restrict_box import threading from tqdm import tqdm from queue import Queue import time # classification from knnmodel import init_KNN, get_data_from_video, init_get_video from edge import offside_dectet from ball_touch import ball_state root_cfg.merge_from_file('./experiments/siamfcpp/siamfcpp_tinyconv.yaml') task, task_cfg = specify_task(root_cfg) task_cfg.freeze() # construct the argument parser and parse the arguments ap = argparse.ArgumentParser() ap.add_argument( "-v", "--video", type=str, default="./video/video5.mp4", #"rtmp://192.168.43.109:9999/live/test" help="path to input video file") ap.add_argument("-t", "--tracker", type=str, default="kcf",