def test_load_config(self): PipelineManager.pipeline_types = { "GStreamer": "DummyType" } expected_result = { 'gstreamer': { 'object_detection': {} }, 'object_detection': { '1': { "name": "object_detection", "version": 1, "type": "GStreamer", "description": "Object Detection Pipeline" } } } expected_max_running_pipelines = 3 PipelineManager.load_config(os.path.join(os.path.dirname(__file__),"pipelines"), expected_max_running_pipelines) result = PipelineManager.pipelines result_max_running_pipelines = PipelineManager.MAX_RUNNING_PIPELINES self.assertEqual(result, expected_result) self.assertEqual(result_max_running_pipelines, expected_max_running_pipelines)
def __init__(self, pipeline, version="2"): super(RunVA, self).__init__() self._pipeline = pipeline self._version = version self._db = DBIngest(host=dbhost, index="algorithms", office=office) self._maincontext = GLib.MainLoop().get_context() self._stop = None ModelManager.load_config("/home/models", {}) PipelineManager.load_config("/home/pipelines", 1)
def main(options): PipelineManager.load_config( os.path.join(CONFIG_PATH, options.pipeline_dir), MAX_RUNNING_PIPELINES) ModelManager.load_config(os.path.join(CONFIG_PATH, options.model_dir)) asyncio.set_event_loop(asyncio.new_event_loop()) app = connexion.App(__name__, specification_dir='./openapi/') app.app.json_encoder = encoder.JSONEncoder app.add_api('openapi.yaml', arguments={'title': 'Video Analytics API'}) logger.info("Starting Tornado Server on port: {p}".format(p=options.port)) app.run(server='tornado', port=options.port)
def main(options): PipelineManager.load_config( os.path.join(options.config_path, options.pipeline_dir), options.max_running_pipelines) ModelManager.load_config( os.path.join(options.config_path, options.model_dir), parse_network_preference(options)) app = connexion.App(__name__, specification_dir='./openapi/') app.app.json_encoder = encoder.JSONEncoder app.add_api('openapi.yaml', arguments={'title': 'Video Analytics Serving API'}) logger.info("Starting Tornado Server on port: {p}".format(p=options.port)) app.run(server='tornado', port=options.port)
def test_load_config(self): PipelineManager.pipeline_types = {"GStreamer": "DummyType"} expected_result = { 'gstreamer': { 'object_detection': {} }, 'object_detection': { '1': { "name": "object_detection", "version": 1, "type": "GStreamer", "description": "Object Detection Pipeline" } } } PipelineManager.load_config("modules/test/pipelines") result = PipelineManager.pipelines self.assertEqual(result, expected_result)
def __init__(self): super(RunVA, self).__init__() ModelManager.load_config("/home/models", {}) PipelineManager.load_config("/home/pipelines", 1) self._maincontext = GLib.MainLoop().get_context() GLib.timeout_add(1000, self._noop)