import platform
from utils.sqlitedao import create_database
from cfg import DataConfig, TaskConfig
from task import TaskManager, ImgInfoPool, get_cfgDataHandler
import importlib


def args_parse():
    parser = argparse.ArgumentParser()
    parser.add_argument('-i', type=str, help='input video')
    return parser.parse_args()


if __name__ == "__main__":
    # Linux平台启动
    if platform.system() == 'Linux':
        mp.set_start_method('spawn', force=True)
    img_info_pool = ImgInfoPool(max_size=30)
    tm = TaskManager(img_info_pool)
    args = args_parse()
    cfg_data = {
        "task_type": "collectTrack",
        "filepath": args.i,
    }
    task_type = cfg_data['task_type']
    get_cfg = importlib.import_module(TaskConfig.SIMPLE_CFG_DIR +
                                      task_type).get_injected_cfg
    task_cfg = get_cfg(cfg_data)
    tm.submit(task_name=os.path.basename(args.i), task_cfg=task_cfg)
    tm.resume(task_name=os.path.basename(args.i), is_join=True)
예제 #2
0
from cfg import DataConfig, TaskConfig
from task import TaskManager, ImgInfoPool, get_cfgDataHandler
import importlib

if __name__ == '__main__':
    # Linux平台启动
    if platform.system() == 'Linux':
        mp.set_start_method('spawn', force=True)
    create_database(True)
    img_info_pool = ImgInfoPool(max_size=30)
    tm = TaskManager(img_info_pool)
    cfg_data = {
        "task_type":
        "crossRoadsTaskFake",
        "filename":
        "lot_15.mp4",
        "parking_monitoring_area": [[[1250, 800], [1250, 400], [750, 400],
                                     [750, 800]]],
        "lane_monitoring_area": [[[1250, 1000], [1250, 300], [500, 300],
                                  [500, 1000]]],
        "lane_no_allow_cars": {
            "2": ["car"]
        }
    }
    task_type = cfg_data['task_type']
    get_cfg = importlib.import_module(TaskConfig.SCENE_CFG_DIR +
                                      task_type).get_injected_cfg
    task_cfg = get_cfg(cfg_data)
    tm.submit(task_name='lot_15.mp4', task_cfg=task_cfg)
    tm.resume(task_name='lot_15.mp4')
from cfg import DataConfig, TaskConfig
from task import TaskManager, ImgInfoPool, get_cfgDataHandler
import importlib

if __name__ == '__main__':
    # Linux平台启动
    if platform.system() == 'Linux':
        mp.set_start_method('spawn', force=True)
    create_database(True)
    img_info_pool = ImgInfoPool(max_size=30)
    tm = TaskManager(img_info_pool)
    cfg_data = {
        "task_type":
        "expresswayTaskFake",
        "filename":
        "gs.mp4",
        "parking_monitoring_area": [[[1250, 800], [1250, 400], [750, 400],
                                     [750, 800]]],
        "lane_monitoring_area": [[[1250, 800], [1250, 400], [750, 400],
                                  [750, 800]]],
        "lane_no_allow_cars": {
            "2": ["car"]
        }
    }
    task_type = cfg_data['task_type']
    get_cfg = importlib.import_module(TaskConfig.SCENE_CFG_DIR +
                                      task_type).get_injected_cfg
    task_cfg = get_cfg(cfg_data)
    tm.submit(task_name='gs.mp4', task_cfg=task_cfg)
    tm.resume(task_name='gs.mp4')
예제 #4
0
from task import TaskManager, ImgInfoPool, get_cfgDataHandler
import importlib


def args_parse():
    parser = argparse.ArgumentParser()
    parser.add_argument('-i', type=str, help='input video')
    parser.add_argument('-o', type=str, help='save path')
    return parser.parse_args()


if __name__ == "__main__":
    # Linux平台启动
    if platform.system() == 'Linux':
        mp.set_start_method('spawn', force=True)
    img_info_pool = ImgInfoPool(max_size=30)
    tm = TaskManager(img_info_pool)
    args = args_parse()
    cfg_data = {
        "task_type": "processVideo",
        "filepath": args.i,
        "write_path": args.o,
        "resolution": (1600, 900)
    }
    task_type = cfg_data['task_type']
    get_cfg = importlib.import_module(TaskConfig.SIMPLE_CFG_DIR +
                                      task_type).get_injected_cfg
    task_cfg = get_cfg(cfg_data)
    tm.submit(task_name=os.path.basename(args.i), task_cfg=task_cfg)
    tm.resume(task_name=os.path.basename(args.i))