feat(template): add image pipeline template

Author: megvii-mge

model.py attr.pymodel.py renamed to attr.py

@@ -9,28 +9,17 @@
 #!/usr/bin/env python
 # coding=utf-8
 
-import json
-import numpy as np
-from loguru import logger
 from megflow import register
 
-from .lite import PredictorLite
-
-
 @register(inputs=['inp'], outputs=['out'])
-class Classify:
+class Attribute:
     def __init__(self, name, arg):
-        logger.info("loading model...")
         self.name = name
-        self._model = PredictorLite(path=arg['path'],
-                                    device=arg['device'],
-                                    device_id=arg['device_id'])
 
     def exec (self):
         envelope = self.inp.recv()
         if envelope is None:
             return
+        print('attr')
+        self.out.send(envelope)
 
-        data = envelope.msg['data']
-        result = self._model.inference(data)
-        self.out.send(envelope.repack(result))

config.toml

@@ -1,21 +1,27 @@
-main = "quickstart-modelserving"
+main = "image"
+
+[[nodes]]
+name = "det"
+ty = "Detect"
+device = "cpu"
+device_id = 0
+
+[[nodes]]
+name = "attr"
+ty = "Attribute"
+device = "cpu"
+device_id = 0
 
 [[graphs]]
 name = "subgraph"
-inputs = [{ name = "inp", cap = 16, ports = ["model:inp"] }]
-outputs = [{ name = "out", cap = 16, ports = ["model:out"] }]
+inputs = [{ name = "inp", cap = 16, ports = ["det:inp"] }]
+outputs = [{ name = "out", cap = 16, ports = ["attr:out"] }]
 connections = [
+  { cap = 16, ports = ["det:out", "attr:inp"] },
 ]
 
-    [[graphs.nodes]]
-    name = "model"
-    ty = "Model"
-    path = "##PATH##"
-    device = "cpu"
-    device_id = 0
-    
 [[graphs]]
-name = "tutorial_01_image"
+name = "image"
 connections = [
   { cap = 16, ports = ["source:out", "destination:inp"] },
   { cap = 16, ports = ["source:inp", "destination:out"] }
@@ -25,7 +31,6 @@ connections = [
     name = "source"
     ty = "ImageServer"
     port = 8080
-    response = "json"
 
     [[graphs.nodes]]
     name = "destination"

det.py

@@ -0,0 +1,27 @@
+# MegFlow is Licensed under the Apache License, Version 2.0 (the "License")
+#
+# Copyright (c) 2019-2021 Megvii Inc. All rights reserved.
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
+#!/usr/bin/env python
+# coding=utf-8
+
+from megflow import register
+
+@register(inputs=['inp'], outputs=['out'])
+class Detect:
+    def __init__(self, name, arg):
+        print('init det')
+        self.name = name
+
+    def exec (self):
+        envelope = self.inp.recv()
+        print('det')
+        if envelope is None:
+            return
+        print('det')
+        self.out.send(envelope)
+