[Feature] Add NCNN on MMSegmentation (open-mmlab#55)

* fix custom ops support, fix multiple mark bug, add name mapping * check if the value_info need to be added * remove unnecessary print * add nms implement * two stage split wip * add two stage split * add split retinanet visualize * add two stage split (wip) * finish two stage split * fix lint * move parse string to mmdeploy.utils * add calib data generator * create calib dataset * finish end2end int8 * add split two stage tensorrt visualize * first * fix0 * fix1 * dirty work * wip * add allocator * finally done! * lint * fix lint * better gather * better onnx2ncnn * fix tensorslice bugs * fix lint * fix clang-format * remove comments * fix expand * int param * fix lint * [Fix] NCNN TensorSlice op bugs (open-mmlab#42) * fix custom ops support, fix multiple mark bug, add name mapping * check if the value_info need to be added * remove unnecessary print * add nms implement * two stage split wip * add two stage split * add split retinanet visualize * add two stage split (wip) * finish two stage split * fix lint * move parse string to mmdeploy.utils * add calib data generator * create calib dataset * finish end2end int8 * add split two stage tensorrt visualize * fix tensorslice bugs * fix lint * fix clang-format * remove comments * int param * fix lint Co-authored-by: grimoire <[email protected]> * add two stage ncnn support * remove unused ops * git unused config * remove no_grad, should add in refactor * add ncnn wrapper * fix lint * size return tuple * Resolve grammar error * Fix lint * Trim Trailing Whitespace * fix trim * add argmax to topk * add ArgMax parse * add ncnn mmseg deploy cfg * utils add ncnn mmseg * add ncnn * fix lint * fix yapf * fix clang-format-9 * remove debugging code Co-authored-by: grimoire <[email protected]> Co-authored-by: grimoire <[email protected]> Co-authored-by: maningsheng <[email protected]>
tpoisonooo · Sep 3, 2021 · 56bc124 · 56bc124
1 parent e73d9fb
commit 56bc124
Show file tree

Hide file tree

Showing 6 changed files with 806 additions and 353 deletions.
diff --git a/backend_ops/ncnn/onnx2ncnn/onnx2ncnn.cpp b/backend_ops/ncnn/onnx2ncnn/onnx2ncnn.cpp
@@ -3296,6 +3296,8 @@ int main(int argc, char** argv) {
       fprintf(pp, "%-16s", "UnaryOp");
     } else if (op == "Add") {
       fprintf(pp, "%-16s", "BinaryOp");
+    } else if (op == "ArgMax") {
+      fprintf(pp, "%-16s", "TopK");
     } else if (op == "Asin") {
       fprintf(pp, "%-16s", "UnaryOp");
     } else if (op == "Atan") {
@@ -3604,6 +3606,11 @@ int main(int argc, char** argv) {
         fprintf(pp, " 1=%d", with_scalar);
         fprintf(pp, " 2=%e", b);
       }
+    } else if (op == "ArgMax") {
+      int axis = get_node_attr_i(node, "axis");
+      int keepdims = get_node_attr_i(node, "keepdims");
+      fprintf(pp, " 0=%d", axis - 1);
+      fprintf(pp, " 3=%d", keepdims);
     } else if (op == "Asin") {
       int op_type = 12;
       fprintf(pp, " 0=%d", op_type);