k20hcmus/FishEye8K

1from functools import wraps2from time import time3 4 5def is_video(ext: str):6    """7    Returns true if ext exists in8    allowed_exts for video files.9 10    Args:11        ext:12 13    Returns:14 15    """16 17    allowed_exts = ('.mp4', '.webm', '.ogg', '.avi', '.wmv', '.mkv', '.3gp')18    return any((ext.endswith(x) for x in allowed_exts))19 20 21def tik_tok(func):22    """23    keep track of time for each process.24    Args:25        func:26 27    Returns:28 29    """30    @wraps(func)31    def _time_it(*args, **kwargs):32        start = time()33        try:34            return func(*args, **kwargs)35        finally:36            end_ = time()37            print("time: {:.03f}s, fps: {:.03f}".format(end_ - start, 1 / (end_ - start)))38 39    return _time_it40