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",
