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)
