Merge pull request #22529 from fengyuentau:scatter_scatternd
authorAlexander Smorkalov <2536374+asmorkalov@users.noreply.github.com>
Mon, 17 Oct 2022 11:57:46 +0000 (14:57 +0300)
committerGitHub <noreply@github.com>
Mon, 17 Oct 2022 11:57:46 +0000 (14:57 +0300)
DNN: supports Scatter and ScatterND from ONNX

1  2 
modules/dnn/include/opencv2/dnn/all_layers.hpp
modules/dnn/src/init.cpp
modules/dnn/src/onnx/onnx_importer.cpp

Simple merge
index e1792d1fdef4adf10d45e75c94e9a50cb051b024,e91534e4097f1cb5b95f49fcaea667163903320c..798f439c5c54194b243322ac9f4a096517ec29ac
@@@ -3785,11 -3779,10 +3838,12 @@@ void ONNXImporter::buildDispatchMap_ONN
      dispatch["DetectionOutput"] = &ONNXImporter::parseDetectionOutput;
      dispatch["CumSum"] = &ONNXImporter::parseCumSum;
      dispatch["SpaceToDepth"] = dispatch["DepthToSpace"] = &ONNXImporter::parseDepthToSpace;
+     dispatch["ScatterElements"] = dispatch["Scatter"] = dispatch["ScatterND"] = &ONNXImporter::parseScatter;
  
      dispatch["Equal"] = dispatch["Greater"] = dispatch["Less"] = dispatch["Pow"] = dispatch["Add"] =
 -            dispatch["Sub"] = dispatch["Mul"] = dispatch["Div"] = &ONNXImporter::parseElementWise;
 +            dispatch["Sub"] = dispatch["Mul"] = dispatch["Div"] = dispatch["GreaterOrEqual"] =
 +            dispatch["LessOrEqual"] = &ONNXImporter::parseElementWise;
 +
      dispatch["Sum"] = dispatch["Min"] = dispatch["Max"] = &ONNXImporter::parseElementWise;
      dispatch["Range"] = &ONNXImporter::parseRange;