test_tf_importer.cpp 76 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943
  1. // This file is part of OpenCV project.
  2. // It is subject to the license terms in the LICENSE file found in the top-level directory
  3. // of this distribution and at http://opencv.org/license.html.
  4. // Copyright (C) 2017-2019, Intel Corporation, all rights reserved.
  5. // Third party copyrights are property of their respective owners.
  6. /*
  7. Test for Tensorflow models loading
  8. */
  9. #include "test_precomp.hpp"
  10. #include "npy_blob.hpp"
  11. #include <opencv2/dnn/layer.details.hpp> // CV_DNN_REGISTER_LAYER_CLASS
  12. #include <opencv2/dnn/utils/debug_utils.hpp>
  13. namespace opencv_test
  14. {
  15. using namespace cv;
  16. using namespace cv::dnn;
  17. template<typename TString>
  18. static std::string _tf(TString filename)
  19. {
  20. return (getOpenCVExtraDir() + "/dnn/") + filename;
  21. }
  22. TEST(Test_TensorFlow, read_inception)
  23. {
  24. Net net;
  25. {
  26. const string model = findDataFile("dnn/tensorflow_inception_graph.pb", false);
  27. net = readNetFromTensorflow(model);
  28. ASSERT_FALSE(net.empty());
  29. }
  30. net.setPreferableBackend(DNN_BACKEND_OPENCV);
  31. Mat sample = imread(_tf("grace_hopper_227.png"));
  32. ASSERT_TRUE(!sample.empty());
  33. Mat input;
  34. resize(sample, input, Size(224, 224));
  35. input -= Scalar::all(117); // mean sub
  36. Mat inputBlob = blobFromImage(input);
  37. net.setInput(inputBlob, "input");
  38. Mat out = net.forward("softmax2");
  39. std::cout << out.dims << std::endl;
  40. }
  41. TEST(Test_TensorFlow, inception_accuracy)
  42. {
  43. Net net;
  44. {
  45. const string model = findDataFile("dnn/tensorflow_inception_graph.pb", false);
  46. net = readNetFromTensorflow(model);
  47. ASSERT_FALSE(net.empty());
  48. }
  49. net.setPreferableBackend(DNN_BACKEND_OPENCV);
  50. Mat sample = imread(_tf("grace_hopper_227.png"));
  51. ASSERT_TRUE(!sample.empty());
  52. Mat inputBlob = blobFromImage(sample, 1.0, Size(224, 224), Scalar(), /*swapRB*/true);
  53. net.setInput(inputBlob, "input");
  54. Mat out = net.forward("softmax2");
  55. Mat ref = blobFromNPY(_tf("tf_inception_prob.npy"));
  56. normAssert(ref, out);
  57. }
  58. static std::string path(const std::string& file)
  59. {
  60. return findDataFile("dnn/tensorflow/" + file);
  61. }
  62. class Test_TensorFlow_layers : public DNNTestLayer
  63. {
  64. public:
  65. void runTensorFlowNet(const std::string& prefix, bool hasText = false,
  66. double l1 = 0.0, double lInf = 0.0, bool memoryLoad = false, const std::string& groupPrefix = "")
  67. {
  68. if (cvtest::debugLevel > 0)
  69. {
  70. std::cout << prefix << groupPrefix << std::endl;
  71. }
  72. std::string netPath = path(prefix + groupPrefix + "_net.pb");
  73. std::string netConfig = (hasText ? path(prefix + groupPrefix + "_net.pbtxt") : "");
  74. std::string inpPath = path(prefix + "_in.npy");
  75. std::string outPath = path(prefix + groupPrefix + "_out.npy");
  76. cv::Mat input = blobFromNPY(inpPath);
  77. cv::Mat ref = blobFromNPY(outPath);
  78. checkBackend(&input, &ref);
  79. Net net;
  80. if (memoryLoad)
  81. {
  82. // Load files into a memory buffers
  83. std::vector<char> dataModel;
  84. readFileContent(netPath, dataModel);
  85. std::vector<char> dataConfig;
  86. if (hasText)
  87. {
  88. readFileContent(netConfig, dataConfig);
  89. }
  90. net = readNetFromTensorflow(dataModel.data(), dataModel.size(),
  91. dataConfig.data(), dataConfig.size());
  92. }
  93. else
  94. net = readNetFromTensorflow(netPath, netConfig);
  95. ASSERT_FALSE(net.empty());
  96. net.setPreferableBackend(backend);
  97. net.setPreferableTarget(target);
  98. net.setInput(input);
  99. cv::Mat output = net.forward();
  100. normAssert(ref, output, "", l1 ? l1 : default_l1, lInf ? lInf : default_lInf);
  101. if (cvtest::debugLevel > 0 || HasFailure())
  102. {
  103. std::cout << "input: " << input.size << std::endl;
  104. std::cout << input.reshape(1, 1) << std::endl;
  105. std::cout << "ref " << ref.size << std::endl;
  106. std::cout << ref.reshape(1, 1) << std::endl;
  107. std::cout << "output: " << output.size << std::endl;
  108. std::cout << output.reshape(1, 1) << std::endl;
  109. }
  110. }
  111. };
  112. TEST_P(Test_TensorFlow_layers, reduce_mean)
  113. {
  114. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  115. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  116. runTensorFlowNet("global_pool_by_axis");
  117. }
  118. TEST_P(Test_TensorFlow_layers, reduce_max)
  119. {
  120. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  121. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  122. runTensorFlowNet("max_pool_by_axis", false, 0.0f, 0.0f);
  123. }
  124. TEST_P(Test_TensorFlow_layers, reduce_sum)
  125. {
  126. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  127. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  128. runTensorFlowNet("sum_pool_by_axis");
  129. }
  130. TEST_P(Test_TensorFlow_layers, reduce_max_channel)
  131. {
  132. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2020040000)
  133. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD) // incorrect result
  134. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  135. #endif
  136. runTensorFlowNet("reduce_max_channel", false, 0.0f, 0.0f);
  137. }
  138. TEST_P(Test_TensorFlow_layers, reduce_sum_channel)
  139. {
  140. runTensorFlowNet("reduce_sum_channel");
  141. }
  142. TEST_P(Test_TensorFlow_layers, reduce_max_channel_keep_dims)
  143. {
  144. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2020040000)
  145. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD) // incorrect result
  146. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  147. #endif
  148. runTensorFlowNet("reduce_max_channel", false, 0.0, 0.0, false, "_keep_dims");
  149. }
  150. TEST_P(Test_TensorFlow_layers, reduce_sum_channel_keep_dims)
  151. {
  152. runTensorFlowNet("reduce_sum_channel", false, 0.0, 0.0, false, "_keep_dims");
  153. }
  154. TEST_P(Test_TensorFlow_layers, ArgLayer)
  155. {
  156. if (backend != DNN_BACKEND_OPENCV || target != DNN_TARGET_CPU)
  157. throw SkipTestException("Only CPU is supported"); // FIXIT use tags
  158. runTensorFlowNet("argmax");
  159. runTensorFlowNet("argmin");
  160. }
  161. TEST_P(Test_TensorFlow_layers, conv_single_conv)
  162. {
  163. runTensorFlowNet("single_conv");
  164. }
  165. TEST_P(Test_TensorFlow_layers, conv_atrous_conv2d_valid)
  166. {
  167. runTensorFlowNet("atrous_conv2d_valid");
  168. }
  169. TEST_P(Test_TensorFlow_layers, conv_atrous_conv2d_same)
  170. {
  171. runTensorFlowNet("atrous_conv2d_same");
  172. }
  173. TEST_P(Test_TensorFlow_layers, conv_depthwise_conv2d)
  174. {
  175. runTensorFlowNet("depthwise_conv2d");
  176. }
  177. TEST_P(Test_TensorFlow_layers, conv_keras_atrous_conv2d_same)
  178. {
  179. runTensorFlowNet("keras_atrous_conv2d_same");
  180. }
  181. TEST_P(Test_TensorFlow_layers, conv_pool_nchw)
  182. {
  183. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  184. // [ GENERAL_ERROR ] AssertionFailed: !expired()
  185. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  186. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  187. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2020020000)
  188. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  189. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  190. #endif
  191. runTensorFlowNet("conv_pool_nchw");
  192. }
  193. TEST_P(Test_TensorFlow_layers, Convolution3D)
  194. {
  195. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  196. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target != DNN_TARGET_CPU)
  197. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER); // Only CPU on DLIE backend is supported
  198. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target != DNN_TARGET_CPU)
  199. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH); // Only CPU on DLIE backend is supported
  200. #endif
  201. runTensorFlowNet("conv3d");
  202. }
  203. TEST_P(Test_TensorFlow_layers, padding)
  204. {
  205. runTensorFlowNet("padding_valid");
  206. runTensorFlowNet("spatial_padding");
  207. runTensorFlowNet("mirror_pad");
  208. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2019020000) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  209. if (target == DNN_TARGET_MYRIAD)
  210. {
  211. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  212. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  213. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  214. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  215. }
  216. #endif
  217. runTensorFlowNet("keras_pad_concat");
  218. }
  219. TEST_P(Test_TensorFlow_layers, padding_asymmetric_1)
  220. {
  221. runTensorFlowNet("conv2d_asymmetric_pads_nchw");
  222. }
  223. TEST_P(Test_TensorFlow_layers, padding_asymmetric_2)
  224. {
  225. runTensorFlowNet("conv2d_asymmetric_pads_nhwc");
  226. }
  227. TEST_P(Test_TensorFlow_layers, padding_asymmetric_3)
  228. {
  229. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  230. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_CPU) // Exception: Unsupported pad value
  231. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_CPU, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  232. #endif
  233. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2020020000)
  234. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD) // Exception: Unsupported pad value
  235. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  236. #endif
  237. runTensorFlowNet("max_pool2d_asymmetric_pads_nchw");
  238. }
  239. TEST_P(Test_TensorFlow_layers, padding_asymmetric_4)
  240. {
  241. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  242. // Unsupported pad value
  243. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  244. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  245. // accuracy
  246. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && (target == DNN_TARGET_OPENCL || target == DNN_TARGET_OPENCL_FP16))
  247. applyTestTag(target == DNN_TARGET_OPENCL ? CV_TEST_TAG_DNN_SKIP_IE_OPENCL : CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16,
  248. CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION
  249. );
  250. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021040000)
  251. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD) // Exception: Unsupported pad value
  252. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  253. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  254. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_CPU) // Exception: Unsupported pad value
  255. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_CPU, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  256. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2020020000)
  257. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD) // Exception: Unsupported pad value
  258. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  259. #endif
  260. #endif
  261. runTensorFlowNet("max_pool2d_asymmetric_pads_nhwc");
  262. }
  263. TEST_P(Test_TensorFlow_layers, padding_asymmetric_5)
  264. {
  265. runTensorFlowNet("conv2d_backprop_input_asymmetric_pads_nchw");
  266. }
  267. TEST_P(Test_TensorFlow_layers, padding_asymmetric_6)
  268. {
  269. runTensorFlowNet("conv2d_backprop_input_asymmetric_pads_nhwc");
  270. }
  271. TEST_P(Test_TensorFlow_layers, padding_same)
  272. {
  273. // Reference output values are in range [0.0006, 2.798]
  274. runTensorFlowNet("padding_same");
  275. }
  276. TEST_P(Test_TensorFlow_layers, eltwise)
  277. {
  278. runTensorFlowNet("eltwise_add_mul");
  279. runTensorFlowNet("eltwise_sub");
  280. }
  281. TEST_P(Test_TensorFlow_layers, eltwise_add_vec)
  282. {
  283. runTensorFlowNet("eltwise_add_vec");
  284. }
  285. TEST_P(Test_TensorFlow_layers, eltwise_mul_vec)
  286. {
  287. runTensorFlowNet("eltwise_mul_vec");
  288. }
  289. TEST_P(Test_TensorFlow_layers, tf_reshape_nhwc)
  290. {
  291. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  292. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  293. runTensorFlowNet("tf_reshape_nhwc");
  294. }
  295. TEST_P(Test_TensorFlow_layers, channel_broadcast)
  296. {
  297. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  298. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  299. runTensorFlowNet("channel_broadcast");
  300. }
  301. TEST_P(Test_TensorFlow_layers, pad_and_concat)
  302. {
  303. runTensorFlowNet("pad_and_concat");
  304. }
  305. TEST_P(Test_TensorFlow_layers, concat_axis_1)
  306. {
  307. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021040000)
  308. // IE Exception: Ngraph operation Transpose with name Flatten_1/flatten/Reshape/nhwc has dynamic output shape on 0 port, but CPU plug-in supports only static shape
  309. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && (target == DNN_TARGET_OPENCL || target == DNN_TARGET_OPENCL_FP16))
  310. applyTestTag(target == DNN_TARGET_OPENCL ? CV_TEST_TAG_DNN_SKIP_IE_OPENCL : CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16,
  311. CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION
  312. );
  313. #endif
  314. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021030000)
  315. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_OPENCL)
  316. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_OPENCL, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH); // exception
  317. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_OPENCL_FP16)
  318. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH); // exception
  319. #endif
  320. runTensorFlowNet("concat_axis_1");
  321. }
  322. TEST_P(Test_TensorFlow_layers, concat_3d)
  323. {
  324. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  325. if (backend == DNN_BACKEND_OPENCV && target != DNN_TARGET_CPU)
  326. {
  327. if (target == DNN_TARGET_OPENCL_FP16) applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16);
  328. if (target == DNN_TARGET_OPENCL) applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_OPENCL);
  329. }
  330. if ((backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH ||
  331. backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019) && target == DNN_TARGET_MYRIAD)
  332. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  333. #endif
  334. runTensorFlowNet("concat_3d");
  335. }
  336. TEST_P(Test_TensorFlow_layers, batch_norm_1)
  337. {
  338. runTensorFlowNet("batch_norm");
  339. }
  340. TEST_P(Test_TensorFlow_layers, batch_norm_2)
  341. {
  342. runTensorFlowNet("batch_norm", false, 0.0, 0.0, true);
  343. }
  344. TEST_P(Test_TensorFlow_layers, batch_norm_3)
  345. {
  346. runTensorFlowNet("fused_batch_norm");
  347. }
  348. TEST_P(Test_TensorFlow_layers, batch_norm_4)
  349. {
  350. runTensorFlowNet("fused_batch_norm", false, 0.0, 0.0, true);
  351. }
  352. TEST_P(Test_TensorFlow_layers, batch_norm_5)
  353. {
  354. runTensorFlowNet("batch_norm_text", true);
  355. }
  356. TEST_P(Test_TensorFlow_layers, batch_norm_6)
  357. {
  358. runTensorFlowNet("batch_norm_text", true, 0.0, 0.0, true);
  359. }
  360. TEST_P(Test_TensorFlow_layers, batch_norm_7)
  361. {
  362. runTensorFlowNet("unfused_batch_norm");
  363. }
  364. TEST_P(Test_TensorFlow_layers, batch_norm_8)
  365. {
  366. runTensorFlowNet("fused_batch_norm_no_gamma");
  367. }
  368. TEST_P(Test_TensorFlow_layers, batch_norm_9)
  369. {
  370. runTensorFlowNet("unfused_batch_norm_no_gamma");
  371. }
  372. TEST_P(Test_TensorFlow_layers, batch_norm_10)
  373. {
  374. runTensorFlowNet("mvn_batch_norm");
  375. }
  376. TEST_P(Test_TensorFlow_layers, batch_norm_11)
  377. {
  378. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021030000)
  379. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_CPU)
  380. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_CPU, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH); // nan
  381. #endif
  382. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  383. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  384. runTensorFlowNet("mvn_batch_norm_1x1");
  385. }
  386. TEST_P(Test_TensorFlow_layers, batch_norm_12)
  387. {
  388. runTensorFlowNet("switch_identity");
  389. }
  390. TEST_P(Test_TensorFlow_layers, batch_norm_13)
  391. {
  392. runTensorFlowNet("keras_batch_norm_training");
  393. }
  394. TEST_P(Test_TensorFlow_layers, batch_norm3D)
  395. {
  396. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target != DNN_TARGET_CPU)
  397. {
  398. if (target == DNN_TARGET_OPENCL_FP16) applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  399. if (target == DNN_TARGET_OPENCL) applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_OPENCL, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  400. if (target == DNN_TARGET_MYRIAD) applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  401. throw SkipTestException("");
  402. }
  403. runTensorFlowNet("batch_norm3d");
  404. }
  405. TEST_P(Test_TensorFlow_layers, slim_batch_norm)
  406. {
  407. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  408. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  409. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  410. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  411. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  412. #endif
  413. // Output values range: [-40.0597, 207.827]
  414. double l1 = default_l1;
  415. double lInf = default_lInf;
  416. if (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD)
  417. {
  418. l1 = 0.041;
  419. lInf = 0.33;
  420. }
  421. #if defined(INF_ENGINE_RELEASE)
  422. else if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_CPU)
  423. {
  424. lInf = 0.0002;
  425. }
  426. #endif
  427. else if (target == DNN_TARGET_CUDA_FP16)
  428. {
  429. l1 = 0.005;
  430. lInf = 0.33;
  431. }
  432. else if (target == DNN_TARGET_CPU_FP16)
  433. {
  434. l1 = 0.041;
  435. lInf = 0.37;
  436. }
  437. runTensorFlowNet("slim_batch_norm", false, l1, lInf);
  438. }
  439. TEST_P(Test_TensorFlow_layers, pooling_max_pool_even)
  440. {
  441. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  442. // [ GENERAL_ERROR ] AssertionFailed: !expired()
  443. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  444. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  445. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2020020000)
  446. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  447. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  448. #endif
  449. runTensorFlowNet("max_pool_even");
  450. }
  451. TEST_P(Test_TensorFlow_layers, pooling_max_pool_odd_valid)
  452. {
  453. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  454. // [ GENERAL_ERROR ] AssertionFailed: !expired()
  455. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  456. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  457. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2020020000)
  458. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  459. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  460. #endif
  461. runTensorFlowNet("max_pool_odd_valid");
  462. }
  463. TEST_P(Test_TensorFlow_layers, pooling_max_pool_odd_same)
  464. {
  465. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  466. // [ GENERAL_ERROR ] AssertionFailed: !expired()
  467. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  468. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  469. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2020020000)
  470. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  471. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  472. #endif
  473. runTensorFlowNet("max_pool_odd_same");
  474. }
  475. TEST_P(Test_TensorFlow_layers, pooling_reduce_mean)
  476. {
  477. runTensorFlowNet("reduce_mean"); // an average pooling over all spatial dimensions.
  478. }
  479. TEST_P(Test_TensorFlow_layers, pooling_reduce_max)
  480. {
  481. runTensorFlowNet("reduce_max"); // a MAX pooling over all spatial dimensions.
  482. }
  483. TEST_P(Test_TensorFlow_layers, pooling_reduce_sum)
  484. {
  485. runTensorFlowNet("reduce_sum"); // a SUM pooling over all spatial dimensions.
  486. }
  487. TEST_P(Test_TensorFlow_layers, pooling_reduce_sum_0_false)
  488. {
  489. runTensorFlowNet("reduce_sum_0_False");
  490. }
  491. TEST_P(Test_TensorFlow_layers, pooling_reduce_sum_1_false)
  492. {
  493. runTensorFlowNet("reduce_sum_1_False");
  494. }
  495. TEST_P(Test_TensorFlow_layers, pooling_reduce_sum_2_false)
  496. {
  497. runTensorFlowNet("reduce_sum_2_False");
  498. }
  499. TEST_P(Test_TensorFlow_layers, pooling_reduce_sum_3_false)
  500. {
  501. runTensorFlowNet("reduce_sum_3_False");
  502. }
  503. TEST_P(Test_TensorFlow_layers, pooling_reduce_sum_1_2_false)
  504. {
  505. #if defined(INF_ENGINE_RELEASE)
  506. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  507. {
  508. default_l1 = 0.01f;
  509. }
  510. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_OPENCL_FP16)
  511. {
  512. default_l1 = 0.01f;
  513. }
  514. #endif
  515. runTensorFlowNet("reduce_sum_1_2_False");
  516. }
  517. TEST_P(Test_TensorFlow_layers, pooling_reduce_sum_0_true)
  518. {
  519. runTensorFlowNet("reduce_sum_0_True");
  520. }
  521. TEST_P(Test_TensorFlow_layers, pooling_reduce_sum_1_true)
  522. {
  523. runTensorFlowNet("reduce_sum_1_True");
  524. }
  525. TEST_P(Test_TensorFlow_layers, pooling_reduce_sum_2_true)
  526. {
  527. runTensorFlowNet("reduce_sum_2_True");
  528. }
  529. TEST_P(Test_TensorFlow_layers, pooling_reduce_sum_3_true)
  530. {
  531. runTensorFlowNet("reduce_sum_3_True");
  532. }
  533. TEST_P(Test_TensorFlow_layers, pooling_reduce_sum_1_2_true)
  534. {
  535. #if defined(INF_ENGINE_RELEASE)
  536. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  537. {
  538. default_l1 = 0.01f;
  539. }
  540. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_OPENCL_FP16)
  541. {
  542. default_l1 = 0.01f;
  543. }
  544. #endif
  545. runTensorFlowNet("reduce_sum_1_2_True");
  546. }
  547. TEST_P(Test_TensorFlow_layers, max_pool_grad)
  548. {
  549. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2023000000)
  550. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  551. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  552. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  553. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  554. #endif
  555. runTensorFlowNet("max_pool_grad");
  556. }
  557. // TODO: fix tests and replace to pooling
  558. TEST_P(Test_TensorFlow_layers, ave_pool_same)
  559. {
  560. // Reference output values are in range [-0.519531, 0.112976]
  561. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2019010000) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  562. if (target == DNN_TARGET_MYRIAD && getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X)
  563. {
  564. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  565. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  566. else if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  567. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  568. }
  569. #endif
  570. runTensorFlowNet("ave_pool_same");
  571. }
  572. TEST_P(Test_TensorFlow_layers, MaxPooling3D)
  573. {
  574. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  575. // IE exception: [ GENERAL_ERROR ] AssertionFailed: !expired()
  576. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  577. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  578. // accuracy
  579. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && (target == DNN_TARGET_OPENCL || target == DNN_TARGET_OPENCL_FP16))
  580. applyTestTag(target == DNN_TARGET_OPENCL ? CV_TEST_TAG_DNN_SKIP_IE_OPENCL : CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16,
  581. CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION
  582. );
  583. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021040000)
  584. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  585. {
  586. // accuracy
  587. if (target == DNN_TARGET_OPENCL || target == DNN_TARGET_OPENCL_FP16)
  588. applyTestTag(target == DNN_TARGET_OPENCL ? CV_TEST_TAG_DNN_SKIP_IE_OPENCL : CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16,
  589. CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION
  590. );
  591. // IE exception: [ GENERAL_ERROR ] AssertionFailed: !expired()
  592. if (target == DNN_TARGET_MYRIAD)
  593. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  594. }
  595. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  596. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target != DNN_TARGET_CPU)
  597. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER); // Only CPU on DLIE backend is supported
  598. else if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target != DNN_TARGET_CPU)
  599. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH); // Only CPU on DLIE backend is supported
  600. #endif
  601. if (backend == DNN_BACKEND_OPENCV && target != DNN_TARGET_CPU)
  602. throw SkipTestException("Only CPU is supported"); // FIXIT use tags
  603. if (backend == DNN_BACKEND_VKCOM)
  604. applyTestTag(CV_TEST_TAG_DNN_SKIP_VULKAN);
  605. runTensorFlowNet("max_pool3d");
  606. }
  607. TEST_P(Test_TensorFlow_layers, AvePooling3D)
  608. {
  609. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  610. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target != DNN_TARGET_CPU)
  611. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER); // Only CPU on DLIE backend is supported
  612. else if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target != DNN_TARGET_CPU)
  613. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH); // Only CPU on DLIE backend is supported
  614. #endif
  615. if (backend == DNN_BACKEND_OPENCV && target != DNN_TARGET_CPU)
  616. throw SkipTestException("Only CPU is supported"); // FIXIT use tags
  617. if (backend == DNN_BACKEND_VKCOM)
  618. applyTestTag(CV_TEST_TAG_DNN_SKIP_VULKAN);
  619. runTensorFlowNet("ave_pool3d");
  620. }
  621. TEST_P(Test_TensorFlow_layers, deconvolution)
  622. {
  623. if (backend == DNN_BACKEND_CUDA)
  624. applyTestTag(CV_TEST_TAG_DNN_SKIP_CUDA);
  625. runTensorFlowNet("deconvolution");
  626. runTensorFlowNet("deconvolution_same");
  627. runTensorFlowNet("deconvolution_stride_2_same");
  628. runTensorFlowNet("deconvolution_adj_pad_valid");
  629. runTensorFlowNet("deconvolution_adj_pad_same");
  630. runTensorFlowNet("keras_deconv_valid");
  631. runTensorFlowNet("keras_deconv_same");
  632. runTensorFlowNet("keras_deconv_same_v2");
  633. }
  634. TEST_P(Test_TensorFlow_layers, matmul)
  635. {
  636. if (backend == DNN_BACKEND_OPENCV && target == DNN_TARGET_OPENCL_FP16)
  637. applyTestTag(CV_TEST_TAG_DNN_SKIP_OPENCL_FP16);
  638. if (backend == DNN_BACKEND_OPENCV && target == DNN_TARGET_CPU_FP16)
  639. applyTestTag(CV_TEST_TAG_DNN_SKIP_CPU_FP16);
  640. runTensorFlowNet("matmul");
  641. runTensorFlowNet("nhwc_transpose_reshape_matmul");
  642. // Reference output values are in range [-5.688, 4.484]
  643. double l1 = target == DNN_TARGET_MYRIAD ? 6.1e-3 : default_l1;
  644. runTensorFlowNet("nhwc_reshape_matmul", false, l1);
  645. runTensorFlowNet("matmul_layout");
  646. runTensorFlowNet("two_inputs_matmul");
  647. }
  648. TEST_P(Test_TensorFlow_layers, batch_matmul)
  649. {
  650. if (backend == DNN_BACKEND_OPENCV && target == DNN_TARGET_OPENCL_FP16)
  651. applyTestTag(CV_TEST_TAG_DNN_SKIP_OPENCL_FP16);
  652. if (backend == DNN_BACKEND_OPENCV && target == DNN_TARGET_CPU_FP16)
  653. applyTestTag(CV_TEST_TAG_DNN_SKIP_CPU_FP16);
  654. runTensorFlowNet("batch_matmul");
  655. }
  656. TEST_P(Test_TensorFlow_layers, square)
  657. {
  658. if (backend == DNN_BACKEND_OPENCV && target == DNN_TARGET_OPENCL_FP16)
  659. applyTestTag(CV_TEST_TAG_DNN_SKIP_OPENCL_FP16);
  660. if (backend == DNN_BACKEND_OPENCV && target == DNN_TARGET_CPU_FP16)
  661. applyTestTag(CV_TEST_TAG_DNN_SKIP_CPU_FP16);
  662. runTensorFlowNet("square");
  663. }
  664. TEST_P(Test_TensorFlow_layers, reshape)
  665. {
  666. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  667. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  668. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  669. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  670. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  671. #endif
  672. runTensorFlowNet("shift_reshape_no_reorder");
  673. runTensorFlowNet("reshape_no_reorder");
  674. runTensorFlowNet("reshape_reduce");
  675. runTensorFlowNet("reshape_as_shape");
  676. }
  677. TEST_P(Test_TensorFlow_layers, flatten)
  678. {
  679. #if defined(INF_ENGINE_RELEASE)
  680. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_MYRIAD
  681. && getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_2
  682. )
  683. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_2, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  684. #endif
  685. runTensorFlowNet("flatten", true);
  686. }
  687. TEST_P(Test_TensorFlow_layers, unfused_flatten)
  688. {
  689. runTensorFlowNet("unfused_flatten");
  690. runTensorFlowNet("unfused_flatten_unknown_batch");
  691. }
  692. TEST_P(Test_TensorFlow_layers, reshape_layer)
  693. {
  694. runTensorFlowNet("reshape_layer");
  695. }
  696. TEST_P(Test_TensorFlow_layers, reshape_nchw)
  697. {
  698. runTensorFlowNet("reshape_nchw");
  699. }
  700. TEST_P(Test_TensorFlow_layers, reshape_conv)
  701. {
  702. runTensorFlowNet("reshape_conv");
  703. }
  704. TEST_P(Test_TensorFlow_layers, leaky_relu)
  705. {
  706. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2018050000)
  707. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_OPENCL)
  708. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_OPENCL, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  709. #endif
  710. runTensorFlowNet("leaky_relu");
  711. runTensorFlowNet("leaky_relu_order1");
  712. runTensorFlowNet("leaky_relu_order2");
  713. runTensorFlowNet("leaky_relu_order3");
  714. }
  715. TEST_P(Test_TensorFlow_layers, l2_normalize)
  716. {
  717. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2019010000)
  718. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_MYRIAD
  719. && getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X
  720. )
  721. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  722. #endif
  723. runTensorFlowNet("l2_normalize");
  724. }
  725. TEST_P(Test_TensorFlow_layers, BiasAdd)
  726. {
  727. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2019010000)
  728. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_MYRIAD
  729. && getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X
  730. )
  731. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  732. #endif
  733. runTensorFlowNet("bias_add_1");
  734. }
  735. TEST_P(Test_TensorFlow_layers, ExpandDims)
  736. {
  737. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021040000)
  738. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_CPU)
  739. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_CPU, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH); // Layout::ANY is broken on CPU
  740. #endif
  741. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2019010000)
  742. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_MYRIAD
  743. && getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X
  744. )
  745. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  746. #endif
  747. runTensorFlowNet("expand_dims_1");
  748. runTensorFlowNet("expand_dims_2");
  749. }
  750. // TODO: fix it and add to l2_normalize
  751. TEST_P(Test_TensorFlow_layers, l2_normalize_3d)
  752. {
  753. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  754. // accuracy
  755. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  756. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  757. #elif defined(INF_ENGINE_RELEASE)
  758. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_MYRIAD)
  759. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  760. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  761. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  762. #endif
  763. runTensorFlowNet("l2_normalize_3d");
  764. }
  765. class Test_TensorFlow_diagnostics : public DNNTestLayer {
  766. public:
  767. Test_TensorFlow_diagnostics()
  768. {
  769. enableModelDiagnostics(true);
  770. skipModelImport(true);
  771. }
  772. ~Test_TensorFlow_diagnostics()
  773. {
  774. enableModelDiagnostics(false);
  775. skipModelImport(false);
  776. }
  777. void runFailingTensorFlowNet(const std::string& prefix, bool hasText = false)
  778. {
  779. std::string netPath = path(prefix + "_net.pb");
  780. std::string netConfig = (hasText ? path(prefix + "_net.pbtxt") : "");
  781. Net net = readNetFromTensorflow(netPath, netConfig);
  782. }
  783. };
  784. TEST_P(Test_TensorFlow_diagnostics, not_implemented_layer)
  785. {
  786. runFailingTensorFlowNet("not_implemented_layer");
  787. }
  788. TEST_P(Test_TensorFlow_diagnostics, broken_parameters)
  789. {
  790. runFailingTensorFlowNet("broken_layer");
  791. }
  792. INSTANTIATE_TEST_CASE_P(/**/, Test_TensorFlow_diagnostics, dnnBackendsAndTargets());
  793. class Test_TensorFlow_nets : public DNNTestLayer {};
  794. TEST_P(Test_TensorFlow_nets, MobileNet_SSD)
  795. {
  796. #if defined(INF_ENGINE_RELEASE)
  797. if (target == DNN_TARGET_MYRIAD)
  798. {
  799. #if INF_ENGINE_VER_MAJOR_GE(2019020000)
  800. if (getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X)
  801. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X,
  802. backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 ?
  803. CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER :
  804. CV_TEST_TAG_DNN_SKIP_IE_NGRAPH,
  805. CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  806. #endif
  807. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  808. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  809. }
  810. #endif
  811. checkBackend();
  812. std::string imgPath = findDataFile("dnn/street.png");
  813. std::string netConfig = findDataFile("dnn/ssd_mobilenet_v1_coco.pbtxt");
  814. std::string netPath = findDataFile("dnn/ssd_mobilenet_v1_coco.pb", false);
  815. Mat inp;
  816. resize(imread(imgPath), inp, Size(300, 300));
  817. inp = blobFromImage(inp, 1.0f / 127.5, Size(), Scalar(127.5, 127.5, 127.5), true);
  818. Mat ref = blobFromNPY(findDataFile("dnn/tensorflow/ssd_mobilenet_v1_coco.detection_out.npy"));
  819. Net net = readNetFromTensorflow(netPath, netConfig);
  820. net.setPreferableBackend(backend);
  821. net.setPreferableTarget(target);
  822. net.setInput(inp);
  823. Mat out = net.forward();
  824. double scoreDiff = default_l1, iouDiff = default_lInf;
  825. if (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD || target == DNN_TARGET_CPU_FP16)
  826. {
  827. scoreDiff = 0.01;
  828. iouDiff = 0.1;
  829. }
  830. else if (target == DNN_TARGET_CUDA_FP16)
  831. {
  832. iouDiff = 0.04;
  833. }
  834. normAssertDetections(ref, out, "", 0.2, scoreDiff, iouDiff);
  835. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_RELEASE >= 2019010000
  836. expectNoFallbacksFromIE(net);
  837. #endif
  838. }
  839. TEST_P(Test_TensorFlow_nets, Inception_v2_SSD)
  840. {
  841. applyTestTag(target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB);
  842. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LE(2019010000)
  843. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_MYRIAD &&
  844. getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X)
  845. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  846. #endif
  847. checkBackend();
  848. Mat img = imread(findDataFile("dnn/street.png"));
  849. std::string proto = findDataFile("dnn/ssd_inception_v2_coco_2017_11_17.pbtxt");
  850. std::string model = findDataFile("dnn/ssd_inception_v2_coco_2017_11_17.pb", false);
  851. Net net = readNetFromTensorflow(model, proto);
  852. Mat blob = blobFromImage(img, 1.0f, Size(300, 300), Scalar(), true, false);
  853. net.setPreferableBackend(backend);
  854. net.setPreferableTarget(target);
  855. if (target == DNN_TARGET_CPU_FP16)
  856. net.enableWinograd(false);
  857. net.setInput(blob);
  858. // Output has shape 1x1xNx7 where N - number of detections.
  859. // An every detection is a vector of values [id, classId, confidence, left, top, right, bottom]
  860. Mat out = net.forward();
  861. Mat ref = (Mat_<float>(5, 7) << 0, 1, 0.90176028, 0.19872092, 0.36311883, 0.26461923, 0.63498729,
  862. 0, 3, 0.93569964, 0.64865261, 0.45906419, 0.80675775, 0.65708131,
  863. 0, 3, 0.75838411, 0.44668293, 0.45907149, 0.49459291, 0.52197015,
  864. 0, 10, 0.95932811, 0.38349164, 0.32528657, 0.40387636, 0.39165527,
  865. 0, 10, 0.93973452, 0.66561931, 0.37841269, 0.68074018, 0.42907384);
  866. double scoreDiff = default_l1, iouDiff = default_lInf;
  867. if (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD || target == DNN_TARGET_CPU_FP16)
  868. {
  869. scoreDiff = 0.0097;
  870. iouDiff = 0.09;
  871. }
  872. else if (target == DNN_TARGET_CUDA_FP16)
  873. {
  874. scoreDiff = 6e-3;
  875. iouDiff = 0.05;
  876. }
  877. normAssertDetections(ref, out, "", 0.5, scoreDiff, iouDiff);
  878. expectNoFallbacksFromIE(net);
  879. }
  880. TEST_P(Test_TensorFlow_nets, MobileNet_v1_SSD)
  881. {
  882. checkBackend();
  883. std::string proto = findDataFile("dnn/ssd_mobilenet_v1_coco_2017_11_17.pbtxt");
  884. std::string model = findDataFile("dnn/ssd_mobilenet_v1_coco_2017_11_17.pb", false);
  885. Net net = readNetFromTensorflow(model, proto);
  886. Mat img = imread(findDataFile("dnn/dog416.png"));
  887. Mat blob = blobFromImage(img, 1.0f, Size(300, 300), Scalar(), true, false);
  888. net.setPreferableBackend(backend);
  889. net.setPreferableTarget(target);
  890. net.setInput(blob);
  891. Mat out = net.forward();
  892. Mat ref = blobFromNPY(findDataFile("dnn/tensorflow/ssd_mobilenet_v1_coco_2017_11_17.detection_out.npy"));
  893. float scoreDiff = 1.5e-5, iouDiff = 1e-3;
  894. float detectionConfThresh = (target == DNN_TARGET_MYRIAD) ? 0.35 : 0.3;
  895. if (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD || target == DNN_TARGET_CPU_FP16)
  896. {
  897. scoreDiff = 0.011;
  898. iouDiff = 0.012;
  899. }
  900. else if (target == DNN_TARGET_CUDA_FP16)
  901. {
  902. scoreDiff = 0.006;
  903. iouDiff = 0.01;
  904. }
  905. #if defined(INF_ENGINE_RELEASE)
  906. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_MYRIAD &&
  907. getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X)
  908. {
  909. scoreDiff = 0.061;
  910. iouDiff = 0.12;
  911. detectionConfThresh = 0.36;
  912. }
  913. #endif
  914. normAssertDetections(ref, out, "", detectionConfThresh, scoreDiff, iouDiff);
  915. expectNoFallbacksFromIE(net);
  916. }
  917. TEST_P(Test_TensorFlow_nets, Faster_RCNN_inception_v2_coco_2018_01_28)
  918. {
  919. applyTestTag(
  920. (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB),
  921. CV_TEST_TAG_LONG,
  922. CV_TEST_TAG_DEBUG_VERYLONG
  923. );
  924. #ifdef INF_ENGINE_RELEASE
  925. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 &&
  926. (INF_ENGINE_VER_MAJOR_LT(2019020000) || target != DNN_TARGET_CPU))
  927. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  928. if (INF_ENGINE_VER_MAJOR_GT(2019030000) &&
  929. backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  930. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  931. #endif
  932. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  933. // segfault: inference-engine/thirdparty/clDNN/src/gpu/detection_output_cpu.cpp:111:
  934. // Assertion `prior_height > 0' failed.
  935. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_OPENCL_FP16)
  936. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  937. #endif
  938. if (backend == DNN_BACKEND_CUDA && target == DNN_TARGET_CUDA_FP16)
  939. applyTestTag(CV_TEST_TAG_DNN_SKIP_CUDA_FP16);
  940. if (target == DNN_TARGET_CPU_FP16)
  941. applyTestTag(CV_TEST_TAG_DNN_SKIP_CPU_FP16);
  942. checkBackend();
  943. double scoresDiff = 1e-5;
  944. double iouDiff = 1e-4;
  945. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 || backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  946. {
  947. scoresDiff = 0.02;
  948. iouDiff = 0.1;
  949. }
  950. std::string name = "faster_rcnn_inception_v2_coco_2018_01_28";
  951. {
  952. std::string proto = findDataFile("dnn/" + name + ".pbtxt");
  953. std::string model = findDataFile("dnn/" + name + ".pb", false);
  954. Net net = readNetFromTensorflow(model, proto);
  955. net.setPreferableBackend(backend);
  956. net.setPreferableTarget(target);
  957. Mat img = imread(findDataFile("dnn/dog416.png"));
  958. Mat blob = blobFromImage(img, 1.0f, Size(800, 600), Scalar(), true, false);
  959. net.setInput(blob);
  960. Mat out = net.forward();
  961. Mat ref = blobFromNPY(findDataFile("dnn/tensorflow/" + name + ".detection_out.npy"));
  962. // accuracy (both OpenCV & IE)
  963. if (target == DNN_TARGET_OPENCL_FP16)
  964. applyTestTag(CV_TEST_TAG_DNN_SKIP_OPENCL_FP16);
  965. if (target == DNN_TARGET_CPU_FP16)
  966. applyTestTag(CV_TEST_TAG_DNN_SKIP_CPU_FP16);
  967. normAssertDetections(ref, out, name.c_str(), 0.3, scoresDiff, iouDiff);
  968. }
  969. }
  970. TEST_P(Test_TensorFlow_nets, Faster_RCNN_resnet50_coco_2018_01_28)
  971. {
  972. applyTestTag(
  973. (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB),
  974. CV_TEST_TAG_LONG,
  975. CV_TEST_TAG_DEBUG_VERYLONG
  976. );
  977. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  978. // [ GENERAL_ERROR ] AssertionFailed: subgraphTopoSortsStep < subgraphs.size()
  979. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  980. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  981. // [ GENERAL_ERROR ] AssertionFailed: subgraphTopoSortsStep < subgraphs.size()
  982. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && (target == DNN_TARGET_OPENCL || target == DNN_TARGET_OPENCL_FP16))
  983. applyTestTag(target == DNN_TARGET_OPENCL ? CV_TEST_TAG_DNN_SKIP_IE_OPENCL : CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16,
  984. CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION
  985. );
  986. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021040000)
  987. // [ GENERAL_ERROR ] AssertionFailed: subgraphTopoSortsStep++ < subgraphs.size()
  988. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  989. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  990. // IE exception: Ngraph operation Transpose with name FirstStageBoxPredictor/ClassPredictor/reshape_1/nhwc has dynamic output shape on 0 port, but CPU plug-in supports only static shape
  991. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && (target == DNN_TARGET_OPENCL || target == DNN_TARGET_OPENCL_FP16))
  992. applyTestTag(target == DNN_TARGET_OPENCL ? CV_TEST_TAG_DNN_SKIP_IE_OPENCL : CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16,
  993. CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION
  994. );
  995. #elif defined(INF_ENGINE_RELEASE)
  996. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 &&
  997. (INF_ENGINE_VER_MAJOR_LT(2019020000) || target != DNN_TARGET_CPU))
  998. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  999. if (INF_ENGINE_VER_MAJOR_GT(2019030000) &&
  1000. backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  1001. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  1002. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  1003. // segfault: inference-engine/thirdparty/clDNN/src/gpu/detection_output_cpu.cpp:111:
  1004. // Assertion `prior_height > 0' failed.
  1005. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_OPENCL_FP16)
  1006. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  1007. #endif
  1008. #endif
  1009. if (backend == DNN_BACKEND_CUDA && target == DNN_TARGET_CUDA_FP16)
  1010. applyTestTag(CV_TEST_TAG_DNN_SKIP_CUDA_FP16);
  1011. checkBackend();
  1012. double scoresDiff = backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 ? 2.9e-5 : 1e-5;
  1013. double iouDiff = 1e-4;
  1014. if (target == DNN_TARGET_CUDA)
  1015. {
  1016. scoresDiff = 0.06;
  1017. iouDiff = 0.08;
  1018. }
  1019. std::string name = "faster_rcnn_resnet50_coco_2018_01_28";
  1020. {
  1021. std::string proto = findDataFile("dnn/" + name + ".pbtxt");
  1022. std::string model = findDataFile("dnn/" + name + ".pb", false);
  1023. Net net = readNetFromTensorflow(model, proto);
  1024. net.setPreferableBackend(backend);
  1025. net.setPreferableTarget(target);
  1026. Mat img = imread(findDataFile("dnn/dog416.png"));
  1027. Mat blob = blobFromImage(img, 1.0f, Size(800, 600), Scalar(), true, false);
  1028. net.setInput(blob);
  1029. Mat out = net.forward();
  1030. Mat ref = blobFromNPY(findDataFile("dnn/tensorflow/" + name + ".detection_out.npy"));
  1031. // accuracy
  1032. if (target == DNN_TARGET_OPENCL_FP16)
  1033. applyTestTag(CV_TEST_TAG_DNN_SKIP_OPENCL_FP16);
  1034. if (target == DNN_TARGET_CPU_FP16)
  1035. applyTestTag(CV_TEST_TAG_DNN_SKIP_CPU_FP16);
  1036. normAssertDetections(ref, out, name.c_str(), 0.3, scoresDiff, iouDiff);
  1037. }
  1038. }
  1039. TEST_P(Test_TensorFlow_nets, MobileNet_v1_SSD_PPN)
  1040. {
  1041. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2018050000)
  1042. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && (target == DNN_TARGET_OPENCL || target == DNN_TARGET_OPENCL_FP16))
  1043. applyTestTag(target == DNN_TARGET_OPENCL ? CV_TEST_TAG_DNN_SKIP_IE_OPENCL : CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16,
  1044. CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1045. #endif
  1046. checkBackend();
  1047. std::string proto = findDataFile("dnn/ssd_mobilenet_v1_ppn_coco.pbtxt");
  1048. std::string model = findDataFile("dnn/ssd_mobilenet_v1_ppn_coco.pb", false);
  1049. Net net = readNetFromTensorflow(model, proto);
  1050. Mat img = imread(findDataFile("dnn/dog416.png"));
  1051. Mat ref = blobFromNPY(findDataFile("dnn/tensorflow/ssd_mobilenet_v1_ppn_coco.detection_out.npy"));
  1052. Mat blob = blobFromImage(img, 1.0f, Size(300, 300), Scalar(), true, false);
  1053. net.setPreferableBackend(backend);
  1054. net.setPreferableTarget(target);
  1055. net.setInput(blob);
  1056. Mat out = net.forward();
  1057. double scoreDiff = 1.1e-5, iouDiff = default_lInf;
  1058. if (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD || target == DNN_TARGET_CPU_FP16)
  1059. {
  1060. scoreDiff = 0.048;
  1061. iouDiff = 0.058;
  1062. }
  1063. else if (target == DNN_TARGET_CUDA_FP16)
  1064. {
  1065. scoreDiff = 0.006;
  1066. iouDiff = 0.05;
  1067. }
  1068. normAssertDetections(ref, out, "", 0.45, scoreDiff, iouDiff);
  1069. expectNoFallbacksFromIE(net);
  1070. }
  1071. TEST_P(Test_TensorFlow_nets, opencv_face_detector_uint8)
  1072. {
  1073. checkBackend();
  1074. std::string proto = findDataFile("dnn/opencv_face_detector.pbtxt");
  1075. std::string model = findDataFile("dnn/opencv_face_detector_uint8.pb", false);
  1076. Net net = readNetFromTensorflow(model, proto);
  1077. Mat img = imread(findDataFile("gpu/lbpcascade/er.png"));
  1078. Mat blob = blobFromImage(img, 1.0, Size(), Scalar(104.0, 177.0, 123.0), false, false);
  1079. net.setPreferableBackend(backend);
  1080. net.setPreferableTarget(target);
  1081. net.setInput(blob);
  1082. // Output has shape 1x1xNx7 where N - number of detections.
  1083. // An every detection is a vector of values [id, classId, confidence, left, top, right, bottom]
  1084. Mat out = net.forward();
  1085. // References are from test for Caffe model.
  1086. Mat ref = (Mat_<float>(6, 7) << 0, 1, 0.99520785, 0.80997437, 0.16379407, 0.87996572, 0.26685631,
  1087. 0, 1, 0.9934696, 0.2831718, 0.50738752, 0.345781, 0.5985168,
  1088. 0, 1, 0.99096733, 0.13629119, 0.24892329, 0.19756334, 0.3310290,
  1089. 0, 1, 0.98977017, 0.23901358, 0.09084064, 0.29902688, 0.1769477,
  1090. 0, 1, 0.97203469, 0.67965847, 0.06876482, 0.73999709, 0.1513494,
  1091. 0, 1, 0.95097077, 0.51901293, 0.45863652, 0.5777427, 0.5347801);
  1092. double scoreDiff = 3.4e-3, iouDiff = 1e-2;
  1093. if (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD || target == DNN_TARGET_CPU_FP16)
  1094. {
  1095. scoreDiff = 4e-3;
  1096. iouDiff = 0.024;
  1097. }
  1098. else if (target == DNN_TARGET_CUDA_FP16)
  1099. {
  1100. scoreDiff = 4e-3;
  1101. iouDiff = 0.02;
  1102. }
  1103. normAssertDetections(ref, out, "", 0.9, scoreDiff, iouDiff);
  1104. expectNoFallbacksFromIE(net);
  1105. }
  1106. // inp = cv.imread('opencv_extra/testdata/cv/ximgproc/sources/08.png')
  1107. // inp = inp[:,:,[2, 1, 0]].astype(np.float32).reshape(1, 512, 512, 3)
  1108. // outs = sess.run([sess.graph.get_tensor_by_name('feature_fusion/Conv_7/Sigmoid:0'),
  1109. // sess.graph.get_tensor_by_name('feature_fusion/concat_3:0')],
  1110. // feed_dict={'input_images:0': inp})
  1111. // scores = np.ascontiguousarray(outs[0].transpose(0, 3, 1, 2))
  1112. // geometry = np.ascontiguousarray(outs[1].transpose(0, 3, 1, 2))
  1113. // np.save('east_text_detection.scores.npy', scores)
  1114. // np.save('east_text_detection.geometry.npy', geometry)
  1115. TEST_P(Test_TensorFlow_nets, EAST_text_detection)
  1116. {
  1117. applyTestTag(
  1118. (target == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_512MB : CV_TEST_TAG_MEMORY_1GB),
  1119. CV_TEST_TAG_DEBUG_VERYLONG
  1120. );
  1121. #if defined(INF_ENGINE_RELEASE)
  1122. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_MYRIAD)
  1123. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  1124. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  1125. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  1126. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_OPENCL_FP16 &&
  1127. (INF_ENGINE_VER_MAJOR_EQ(2019020000) || INF_ENGINE_VER_MAJOR_GE(2020010000))
  1128. )
  1129. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1130. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_OPENCL_FP16)
  1131. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1132. #endif
  1133. checkBackend();
  1134. std::string netPath = findDataFile("dnn/frozen_east_text_detection.pb", false);
  1135. std::string imgPath = findDataFile("cv/ximgproc/sources/08.png");
  1136. std::string refScoresPath = findDataFile("dnn/east_text_detection.scores.npy");
  1137. std::string refGeometryPath = findDataFile("dnn/east_text_detection.geometry.npy");
  1138. Net net = readNet(netPath);
  1139. net.setPreferableBackend(backend);
  1140. net.setPreferableTarget(target);
  1141. if (target == DNN_TARGET_CPU_FP16)
  1142. net.enableWinograd(false);
  1143. Mat img = imread(imgPath);
  1144. Mat inp = blobFromImage(img, 1.0, Size(), Scalar(123.68, 116.78, 103.94), true, false);
  1145. net.setInput(inp);
  1146. std::vector<Mat> outs;
  1147. std::vector<String> outNames(2);
  1148. outNames[0] = "feature_fusion/Conv_7/Sigmoid";
  1149. outNames[1] = "feature_fusion/concat_3";
  1150. net.forward(outs, outNames);
  1151. Mat scores = outs[0];
  1152. Mat geometry = outs[1];
  1153. // Scores are in range [0, 1]. Geometry values are in range [-0.23, 290]
  1154. double l1_scores = default_l1, lInf_scores = default_lInf;
  1155. double l1_geometry = default_l1, lInf_geometry = default_lInf;
  1156. if (target == DNN_TARGET_OPENCL_FP16)
  1157. {
  1158. lInf_scores = backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 ? 0.16 : 0.11;
  1159. l1_geometry = 0.28; lInf_geometry = 5.94;
  1160. }
  1161. else if (target == DNN_TARGET_MYRIAD)
  1162. {
  1163. lInf_scores = 0.41;
  1164. l1_geometry = 0.28; lInf_geometry = 5.94;
  1165. }
  1166. else if (target == DNN_TARGET_CUDA_FP16)
  1167. {
  1168. lInf_scores = 0.1;
  1169. l1_geometry = 0.3; lInf_geometry = 7;
  1170. }
  1171. else if (target == DNN_TARGET_CPU_FP16)
  1172. {
  1173. lInf_scores = 0.17;
  1174. l1_geometry = 0.28;
  1175. lInf_geometry = 5.94;
  1176. }
  1177. else
  1178. {
  1179. l1_geometry = 1e-4, lInf_geometry = 4.3e-3;
  1180. }
  1181. normAssert(scores, blobFromNPY(refScoresPath), "scores", l1_scores, lInf_scores);
  1182. normAssert(geometry, blobFromNPY(refGeometryPath), "geometry", l1_geometry, lInf_geometry);
  1183. expectNoFallbacksFromIE(net);
  1184. }
  1185. INSTANTIATE_TEST_CASE_P(/**/, Test_TensorFlow_nets, dnnBackendsAndTargets());
  1186. TEST_P(Test_TensorFlow_layers, fp16_weights_fp16_single_conv)
  1187. {
  1188. float l1 = 0.00078, lInf = 0.012;
  1189. runTensorFlowNet("fp16_single_conv", false, l1, lInf);
  1190. }
  1191. TEST_P(Test_TensorFlow_layers, fp16_weights_fp16_max_pool_odd_same)
  1192. {
  1193. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  1194. // [ GENERAL_ERROR ] AssertionFailed: !expired()
  1195. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  1196. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1197. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2020020000)
  1198. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  1199. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1200. #endif
  1201. float l1 = 0.00078, lInf = 0.012;
  1202. runTensorFlowNet("fp16_max_pool_odd_same", false, l1, lInf);
  1203. }
  1204. TEST_P(Test_TensorFlow_layers, fp16_weights_fp16_eltwise_add_mul)
  1205. {
  1206. float l1 = 0.00078, lInf = 0.012;
  1207. runTensorFlowNet("fp16_eltwise_add_mul", false, l1, lInf);
  1208. }
  1209. TEST_P(Test_TensorFlow_layers, fp16_weights_fp16_pad_and_concat)
  1210. {
  1211. float l1 = 0.00078, lInf = 0.012;
  1212. runTensorFlowNet("fp16_pad_and_concat", false, l1, lInf);
  1213. }
  1214. TEST_P(Test_TensorFlow_layers, fp16_weights_fp16_padding_valid)
  1215. {
  1216. float l1 = 0.00078, lInf = 0.012;
  1217. if (target == DNN_TARGET_CPU_FP16)
  1218. l1 = 0.00083;
  1219. runTensorFlowNet("fp16_padding_valid", false, l1, lInf);
  1220. }
  1221. TEST_P(Test_TensorFlow_layers, fp16_weights_fp16_max_pool_even)
  1222. {
  1223. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  1224. // [ GENERAL_ERROR ] AssertionFailed: !expired()
  1225. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  1226. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1227. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2020020000)
  1228. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  1229. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1230. #endif
  1231. float l1 = 0.00078, lInf = 0.012;
  1232. // Reference output values are in range [0.0889, 1.651]
  1233. runTensorFlowNet("fp16_max_pool_even", false, (target == DNN_TARGET_MYRIAD) ? 0.003 : l1, lInf);
  1234. }
  1235. TEST_P(Test_TensorFlow_layers, fp16_weights_fp16_deconvolution)
  1236. {
  1237. float l1 = 0.00078, lInf = 0.012;
  1238. if (target == DNN_TARGET_MYRIAD) {
  1239. l1 = 0.0041;
  1240. lInf = 0.024;
  1241. }
  1242. // Reference output values are in range [0, 10.75]
  1243. runTensorFlowNet("fp16_deconvolution", false, l1, lInf);
  1244. }
  1245. TEST_P(Test_TensorFlow_layers, fp16_weights_fp16_max_pool_odd_valid)
  1246. {
  1247. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  1248. // [ GENERAL_ERROR ] AssertionFailed: !expired()
  1249. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  1250. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1251. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2020020000)
  1252. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  1253. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1254. #endif
  1255. float l1 = 0.00078, lInf = 0.012;
  1256. if (target == DNN_TARGET_MYRIAD) {
  1257. l1 = 0.0041;
  1258. lInf = 0.024;
  1259. }
  1260. // Reference output values are in range [0.418, 2.297]
  1261. runTensorFlowNet("fp16_max_pool_odd_valid", false, l1, lInf);
  1262. }
  1263. TEST_P(Test_TensorFlow_layers, fp16_padding_same)
  1264. {
  1265. float l1 = 7e-4, lInf = 4e-3;
  1266. if (target == DNN_TARGET_CPU_FP16)
  1267. lInf = 5e-3;
  1268. // Reference output values are in range [-3.504, -0.002]
  1269. runTensorFlowNet("fp16_padding_same", false, l1, lInf);
  1270. }
  1271. TEST_P(Test_TensorFlow_layers, defun)
  1272. {
  1273. runTensorFlowNet("defun_dropout");
  1274. }
  1275. TEST_P(Test_TensorFlow_layers, quantized)
  1276. {
  1277. runTensorFlowNet("uint8_single_conv");
  1278. }
  1279. TEST_P(Test_TensorFlow_layers, lstm)
  1280. {
  1281. if(backend == DNN_BACKEND_CUDA)
  1282. applyTestTag(CV_TEST_TAG_DNN_SKIP_CUDA); /* not supported */
  1283. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  1284. // Xlink, Failed to allocate graph: NC_ERROR
  1285. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  1286. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1287. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021040000)
  1288. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  1289. {
  1290. // Exception: Ngraph operation Reshape with name Reshape has dynamic output shape on 0 port, but CPU plug-in supports only static shape
  1291. if (target == DNN_TARGET_OPENCL || target == DNN_TARGET_OPENCL_FP16)
  1292. applyTestTag(target == DNN_TARGET_OPENCL ? CV_TEST_TAG_DNN_SKIP_IE_OPENCL : CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16,
  1293. CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION
  1294. );
  1295. // Xlink
  1296. if (target == DNN_TARGET_MYRIAD)
  1297. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1298. }
  1299. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  1300. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  1301. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  1302. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  1303. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  1304. #endif
  1305. if (backend == DNN_BACKEND_OPENCV && target == DNN_TARGET_OPENCL_FP16)
  1306. applyTestTag(CV_TEST_TAG_DNN_SKIP_OPENCL_FP16);
  1307. if (backend == DNN_BACKEND_OPENCV && target == DNN_TARGET_CPU_FP16)
  1308. applyTestTag(CV_TEST_TAG_DNN_SKIP_CPU_FP16);
  1309. runTensorFlowNet("lstm", true);
  1310. runTensorFlowNet("lstm", true, 0.0, 0.0, true);
  1311. }
  1312. TEST_P(Test_TensorFlow_layers, split)
  1313. {
  1314. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_MYRIAD)
  1315. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  1316. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2023000000)
  1317. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  1318. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  1319. #endif
  1320. runTensorFlowNet("split");
  1321. }
  1322. TEST_P(Test_TensorFlow_layers, split_equals)
  1323. {
  1324. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2023000000)
  1325. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  1326. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  1327. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  1328. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  1329. #endif
  1330. runTensorFlowNet("split_equals");
  1331. }
  1332. TEST_P(Test_TensorFlow_layers, resize_nearest_neighbor)
  1333. {
  1334. runTensorFlowNet("resize_nearest_neighbor");
  1335. }
  1336. TEST_P(Test_TensorFlow_layers, resize_nearest_neighbor_keras_upsampling2d)
  1337. {
  1338. runTensorFlowNet("keras_upsampling2d");
  1339. }
  1340. TEST_P(Test_TensorFlow_layers, resize_nearest_neighbor_align_corners)
  1341. {
  1342. runTensorFlowNet("resize_nearest_neighbor", false, 0.0, 0.0, false, "_align_corners");
  1343. }
  1344. TEST_P(Test_TensorFlow_layers, resize_nearest_neighbor_half_pixel)
  1345. {
  1346. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  1347. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  1348. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  1349. #endif
  1350. runTensorFlowNet("resize_nearest_neighbor", false, 0.0, 0.0, false, "_half_pixel");
  1351. }
  1352. TEST_P(Test_TensorFlow_layers, fused_resize_conv)
  1353. {
  1354. runTensorFlowNet("fused_resize_conv");
  1355. }
  1356. TEST_P(Test_TensorFlow_layers, slice_crop2d)
  1357. {
  1358. double l1 = target == DNN_TARGET_MYRIAD ? 4.9e-3 : default_l1;
  1359. runTensorFlowNet("crop2d", false, l1);
  1360. }
  1361. TEST_P(Test_TensorFlow_layers, slice_4d)
  1362. {
  1363. runTensorFlowNet("slice_4d");
  1364. }
  1365. TEST_P(Test_TensorFlow_layers, slice_strided)
  1366. {
  1367. runTensorFlowNet("strided_slice");
  1368. }
  1369. TEST_P(Test_TensorFlow_layers, softmax_keras)
  1370. {
  1371. runTensorFlowNet("keras_softmax");
  1372. }
  1373. TEST_P(Test_TensorFlow_layers, softmax_slim)
  1374. {
  1375. runTensorFlowNet("slim_softmax");
  1376. }
  1377. TEST_P(Test_TensorFlow_layers, softmax_slim_v2)
  1378. {
  1379. #if defined(INF_ENGINE_RELEASE)
  1380. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_MYRIAD &&
  1381. getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_2
  1382. )
  1383. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_2, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  1384. #endif
  1385. runTensorFlowNet("slim_softmax_v2");
  1386. }
  1387. TEST_P(Test_TensorFlow_layers, relu6)
  1388. {
  1389. runTensorFlowNet("keras_relu6");
  1390. runTensorFlowNet("keras_relu6", /*hasText*/ true);
  1391. }
  1392. TEST_P(Test_TensorFlow_layers, subpixel)
  1393. {
  1394. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2023000000)
  1395. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  1396. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  1397. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  1398. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  1399. #endif
  1400. runTensorFlowNet("subpixel");
  1401. }
  1402. TEST_P(Test_TensorFlow_layers, keras_mobilenet_head)
  1403. {
  1404. runTensorFlowNet("keras_mobilenet_head");
  1405. runTensorFlowNet("keras_learning_phase");
  1406. }
  1407. // TF case: align_corners=False, half_pixel_centers=False
  1408. TEST_P(Test_TensorFlow_layers, resize_bilinear)
  1409. {
  1410. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021030000)
  1411. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  1412. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH); // exception
  1413. #endif
  1414. runTensorFlowNet("resize_bilinear");
  1415. }
  1416. // TF case: align_corners=True, half_pixel_centers=False
  1417. TEST_P(Test_TensorFlow_layers, resize_bilinear_align_corners)
  1418. {
  1419. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021030000)
  1420. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_MYRIAD)
  1421. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH); // exception
  1422. #endif
  1423. runTensorFlowNet("resize_bilinear",
  1424. false, 0.0, 0.0, false, // default parameters
  1425. "_align_corners");
  1426. }
  1427. // TF case: align_corners=False, half_pixel_centers=True
  1428. TEST_P(Test_TensorFlow_layers, resize_bilinear_half_pixel)
  1429. {
  1430. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2023000000)
  1431. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  1432. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  1433. #endif
  1434. runTensorFlowNet("resize_bilinear", false, 0.0, 0.0, false, "_half_pixel");
  1435. }
  1436. // TF case: align_corners=False, half_pixel_centers=False
  1437. TEST_P(Test_TensorFlow_layers, resize_bilinear_factor)
  1438. {
  1439. runTensorFlowNet("resize_bilinear_factor");
  1440. }
  1441. // TF case: align_corners=False, half_pixel_centers=True
  1442. TEST_P(Test_TensorFlow_layers, resize_bilinear_factor_half_pixel)
  1443. {
  1444. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2023000000)
  1445. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  1446. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  1447. #endif
  1448. runTensorFlowNet("resize_bilinear_factor", false, 0.0, 0.0, false, "_half_pixel");
  1449. }
  1450. // TF case: align_corners=True, half_pixel_centers=False
  1451. TEST_P(Test_TensorFlow_layers, resize_bilinear_factor_align_corners)
  1452. {
  1453. runTensorFlowNet("resize_bilinear_factor", false, 0.0, 0.0, false, "_align_corners");
  1454. }
  1455. // TF case: align_corners=False, half_pixel_centers=False
  1456. TEST_P(Test_TensorFlow_layers, resize_bilinear_down)
  1457. {
  1458. runTensorFlowNet("resize_bilinear_down");
  1459. }
  1460. TEST_P(Test_TensorFlow_layers, resize_concat_optimization)
  1461. {
  1462. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021040000)
  1463. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target != DNN_TARGET_CPU) // Exception: Function contains several inputs and outputs with one friendly name! (HETERO bug?)
  1464. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1465. #endif
  1466. runTensorFlowNet("resize_concat_optimization");
  1467. }
  1468. TEST_P(Test_TensorFlow_layers, tf2_dense)
  1469. {
  1470. runTensorFlowNet("tf2_dense");
  1471. }
  1472. TEST_P(Test_TensorFlow_layers, clip_by_value)
  1473. {
  1474. runTensorFlowNet("clip_by_value");
  1475. }
  1476. TEST_P(Test_TensorFlow_layers, tf2_prelu)
  1477. {
  1478. double l1 = 0, lInf = 0;
  1479. if (backend == DNN_BACKEND_CUDA)
  1480. applyTestTag(CV_TEST_TAG_DNN_SKIP_CUDA); // not supported; only across channels is supported
  1481. #if defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2022010000)
  1482. // Eltwise executor got invalid input/output dims configuration
  1483. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_CPU)
  1484. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_CPU, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1485. // Input prelu:StatefulPartitionedCall/StatefulPartitionedCall/sequential/p_re_lu/add hasn't been found in primitiveIDs map
  1486. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && (target == DNN_TARGET_OPENCL || target == DNN_TARGET_OPENCL_FP16))
  1487. applyTestTag(target == DNN_TARGET_OPENCL ? CV_TEST_TAG_DNN_SKIP_IE_OPENCL : CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16,
  1488. CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION
  1489. );
  1490. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_GE(2023000000)
  1491. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH && target == DNN_TARGET_OPENCL) {
  1492. l1 = 1e-4;
  1493. lInf = 1e-3;
  1494. }
  1495. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_EQ(2021040000)
  1496. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  1497. {
  1498. // IE exception: Input prelu:StatefulPartitionedCall/StatefulPartitionedCall/sequential/p_re_lu/add hasn't been found in primitiveIDs map
  1499. if (target == DNN_TARGET_OPENCL || target == DNN_TARGET_OPENCL_FP16)
  1500. applyTestTag(target == DNN_TARGET_OPENCL ? CV_TEST_TAG_DNN_SKIP_IE_OPENCL : CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16,
  1501. CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION
  1502. );
  1503. // IE exception: Eltwise node with name `StatefulPartitionedCall/StatefulPartitionedCall/sequential/p_re_lu/add` has invalid input/output dims configuration
  1504. if (target == DNN_TARGET_CPU)
  1505. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_CPU, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
  1506. }
  1507. #elif defined(INF_ENGINE_RELEASE) && INF_ENGINE_VER_MAJOR_LT(2021040000)
  1508. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
  1509. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  1510. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  1511. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  1512. #endif
  1513. runTensorFlowNet("tf2_prelu", false, l1, lInf);
  1514. }
  1515. TEST_P(Test_TensorFlow_layers, tf2_permute_nhwc_ncwh)
  1516. {
  1517. runTensorFlowNet("tf2_permute_nhwc_ncwh");
  1518. }
  1519. TEST_P(Test_TensorFlow_layers, squeeze)
  1520. {
  1521. #if defined(INF_ENGINE_RELEASE)
  1522. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_MYRIAD
  1523. && getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_2
  1524. )
  1525. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_2, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  1526. #endif
  1527. int inpShapes[][4] = {{1, 3, 4, 2}, {1, 3, 1, 2}, {1, 3, 4, 1}, {1, 3, 4, 1}}; // TensorFlow's shape (NHWC)
  1528. int outShapes[][3] = {{3, 4, 2}, {1, 3, 2}, {1, 3, 4}, {1, 3, 4}};
  1529. int squeeze_dims[] = {0, 2, 3, -1};
  1530. for (int i = 0; i < 4; ++i)
  1531. {
  1532. SCOPED_TRACE(format("i=%d", i));
  1533. std::string pbtxt =
  1534. "node { name: \"input\" op: \"Placeholder\""
  1535. "attr { key: \"data_format\" value { s: \"NHWC\" } } }"
  1536. "node { name: \"squeeze\" op: \"Squeeze\" input: \"input\""
  1537. "attr { key: \"squeeze_dims\" value { list { i:" + format("%d", squeeze_dims[i]) + "}}}}";
  1538. Net net = readNetFromTensorflow(0, 0, pbtxt.c_str(), pbtxt.size());
  1539. net.setPreferableBackend(backend);
  1540. net.setPreferableTarget(target);
  1541. Mat tfInp(4, &inpShapes[i][0], CV_32F);
  1542. randu(tfInp, -1, 1);
  1543. // NHWC to NCHW
  1544. CV_Assert(inpShapes[i][0] == 1);
  1545. std::swap(inpShapes[i][2], inpShapes[i][3]);
  1546. std::swap(inpShapes[i][1], inpShapes[i][2]);
  1547. Mat cvInp = tfInp.reshape(1, tfInp.total() / inpShapes[i][1]).t();
  1548. cvInp = cvInp.reshape(1, 4, &inpShapes[i][0]);
  1549. net.setInput(cvInp);
  1550. Mat out = net.forward();
  1551. normAssert(tfInp.reshape(1, 3, &outShapes[i][0]), out, "", default_l1, default_lInf);
  1552. }
  1553. }
  1554. INSTANTIATE_TEST_CASE_P(/**/, Test_TensorFlow_layers, dnnBackendsAndTargets());
  1555. TEST(Test_TensorFlow, two_inputs)
  1556. {
  1557. Net net = readNet(path("two_inputs_net.pbtxt"));
  1558. net.setPreferableBackend(DNN_BACKEND_OPENCV);
  1559. Mat firstInput(2, 3, CV_32FC1), secondInput(2, 3, CV_32FC1);
  1560. randu(firstInput, -1, 1);
  1561. randu(secondInput, -1, 1);
  1562. net.setInput(firstInput, "first_input");
  1563. net.setInput(secondInput, "second_input");
  1564. Mat out = net.forward();
  1565. normAssert(out, firstInput + secondInput);
  1566. }
  1567. TEST_P(Test_TensorFlow_nets, Mask_RCNN)
  1568. {
  1569. static const double kMaskThreshold = 0.5;
  1570. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_MYRIAD)
  1571. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
  1572. if (target == DNN_TARGET_MYRIAD && getInferenceEngineVPUType() == CV_DNN_INFERENCE_ENGINE_VPU_TYPE_MYRIAD_X)
  1573. applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_X, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
  1574. if (target == DNN_TARGET_CUDA_FP16)
  1575. applyTestTag(CV_TEST_TAG_DNN_SKIP_CUDA_FP16);
  1576. applyTestTag(
  1577. CV_TEST_TAG_MEMORY_2GB,
  1578. CV_TEST_TAG_DEBUG_VERYLONG
  1579. );
  1580. Mat img = imread(findDataFile("dnn/street.png"));
  1581. std::string proto = findDataFile("dnn/mask_rcnn_inception_v2_coco_2018_01_28.pbtxt");
  1582. std::string model = findDataFile("dnn/mask_rcnn_inception_v2_coco_2018_01_28.pb", false);
  1583. Net net = readNetFromTensorflow(model, proto);
  1584. Mat refDetections = blobFromNPY(path("mask_rcnn_inception_v2_coco_2018_01_28.detection_out.npy"));
  1585. Mat refMasks = blobFromNPY(path("mask_rcnn_inception_v2_coco_2018_01_28.detection_masks.npy"));
  1586. Mat blob = blobFromImage(img, 1.0f, Size(800, 800), Scalar(), true, false);
  1587. net.setPreferableBackend(backend);
  1588. net.setPreferableTarget(target);
  1589. if (target == DNN_TARGET_CPU_FP16)
  1590. net.enableWinograd(false);
  1591. net.setInput(blob);
  1592. // Mask-RCNN predicts bounding boxes and segmentation masks.
  1593. std::vector<String> outNames(2);
  1594. outNames[0] = "detection_out_final";
  1595. outNames[1] = "detection_masks";
  1596. std::vector<Mat> outs;
  1597. net.forward(outs, outNames);
  1598. Mat outDetections = outs[0];
  1599. Mat outMasks = outs[1];
  1600. double scoreDiff = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD || target == DNN_TARGET_CPU_FP16) ? 0.2 : 2e-5;
  1601. double iouDiff = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD || target == DNN_TARGET_CPU_FP16) ? 0.018 : default_lInf;
  1602. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  1603. {
  1604. scoreDiff = std::max(scoreDiff, 0.06);
  1605. iouDiff = std::max(iouDiff, 0.01);
  1606. }
  1607. normAssertDetections(refDetections, outDetections, "", /*threshold for zero confidence*/1e-5, scoreDiff, iouDiff);
  1608. // Output size of masks is NxCxHxW where
  1609. // N - number of detected boxes
  1610. // C - number of classes (excluding background)
  1611. // HxW - segmentation shape
  1612. const int numDetections = outDetections.size[2];
  1613. int masksSize[] = {1, numDetections, outMasks.size[2], outMasks.size[3]};
  1614. Mat masks(4, &masksSize[0], CV_32F);
  1615. std::vector<cv::Range> srcRanges(4, cv::Range::all());
  1616. std::vector<cv::Range> dstRanges(4, cv::Range::all());
  1617. outDetections = outDetections.reshape(1, outDetections.total() / 7);
  1618. for (int i = 0; i < numDetections; ++i)
  1619. {
  1620. // Get a class id for this bounding box and copy mask only for that class.
  1621. int classId = static_cast<int>(outDetections.at<float>(i, 1));
  1622. srcRanges[0] = dstRanges[1] = cv::Range(i, i + 1);
  1623. srcRanges[1] = cv::Range(classId, classId + 1);
  1624. outMasks(srcRanges).copyTo(masks(dstRanges));
  1625. }
  1626. cv::Range topRefMasks[] = {Range::all(), Range(0, numDetections), Range::all(), Range::all()};
  1627. refMasks = refMasks(&topRefMasks[0]);
  1628. // make binary masks
  1629. cv::threshold(masks.reshape(1, 1), masks, kMaskThreshold, 1, THRESH_BINARY);
  1630. cv::threshold(refMasks.reshape(1, 1), refMasks, kMaskThreshold, 1, THRESH_BINARY);
  1631. double inter = cv::countNonZero(masks & refMasks);
  1632. double area = cv::countNonZero(masks | refMasks);
  1633. EXPECT_GE(inter / area, (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD || target == DNN_TARGET_CPU_FP16) ? 0.98 : 0.99);
  1634. if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
  1635. expectNoFallbacks(net);
  1636. }
  1637. TEST_P(Test_TensorFlow_nets, EfficientDet)
  1638. {
  1639. if (target != DNN_TARGET_CPU)
  1640. {
  1641. if (target == DNN_TARGET_CPU_FP16) applyTestTag(CV_TEST_TAG_DNN_SKIP_CPU_FP16);
  1642. if (target == DNN_TARGET_OPENCL_FP16) applyTestTag(CV_TEST_TAG_DNN_SKIP_OPENCL_FP16);
  1643. if (target == DNN_TARGET_OPENCL) applyTestTag(CV_TEST_TAG_DNN_SKIP_OPENCL);
  1644. if (target == DNN_TARGET_MYRIAD) applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_MYRIAD);
  1645. }
  1646. checkBackend();
  1647. std::string proto = findDataFile("dnn/efficientdet-d0.pbtxt");
  1648. std::string model = findDataFile("dnn/efficientdet-d0.pb", false);
  1649. Net net = readNetFromTensorflow(model, proto);
  1650. Mat img = imread(findDataFile("dnn/dog416.png"));
  1651. Mat blob = blobFromImage(img, 1.0/255, Size(512, 512), Scalar(123.675, 116.28, 103.53));
  1652. net.setPreferableBackend(backend);
  1653. net.setPreferableTarget(target);
  1654. net.setInput(blob);
  1655. // Output has shape 1x1xNx7 where N - number of detections.
  1656. // An every detection is a vector of values [id, classId, confidence, left, top, right, bottom]
  1657. Mat out = net.forward();
  1658. // References are from test for TensorFlow model.
  1659. Mat ref = (Mat_<float>(3, 7) << 0, 1, 0.8437444, 0.153996080160141, 0.20534580945968628, 0.7463544607162476, 0.7414066195487976,
  1660. 0, 17, 0.8245924, 0.16657517850399017, 0.3996818959712982, 0.4111558794975281, 0.9306337833404541,
  1661. 0, 7, 0.8039304, 0.6118435263633728, 0.13175517320632935, 0.9065558314323425, 0.2943994700908661);
  1662. double scoreDiff = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) ? 4e-3 : 1e-5;
  1663. double iouDiff = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) ? 2e-3 : 1e-4;
  1664. if (target == DNN_TARGET_CUDA_FP16)
  1665. {
  1666. scoreDiff = 0.002;
  1667. iouDiff = 0.005;
  1668. }
  1669. normAssertDetections(ref, out, "", 0.5, scoreDiff, iouDiff);
  1670. expectNoFallbacksFromIE(net);
  1671. }
  1672. TEST(Test_TensorFlow_Importer, tf_graph_simplifier_buffer_overflow_21852)
  1673. {
  1674. uint8_t payload[] = {0x08, 0x08, 0x0a, 0x00, 0x0a, 0x00};
  1675. EXPECT_ANY_THROW(readNetFromTensorflow(reinterpret_cast<const char*>(payload), sizeof(payload) / sizeof(payload[0])));
  1676. }
  1677. // can be triggered with -fsanitize=address
  1678. TEST(Test_TensorFlow_Importer, tf_graph_simplifier_buffer_overflow_21947)
  1679. {
  1680. uint8_t payload[] = {0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00,
  1681. 0xba, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00,
  1682. 0x0a, 0xbd, 0x00, 0x1a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0xba,
  1683. 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00,
  1684. 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0xba, 0x0a, 0x00,
  1685. 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0xba,
  1686. 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00,
  1687. 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x2a, 0x00, 0xba, 0x0a, 0x00,
  1688. 0x0a, 0x00, 0x5d, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x00, 0x0a, 0x40};
  1689. EXPECT_ANY_THROW(readNetFromTensorflow(reinterpret_cast<const char*>(payload), sizeof(payload) / sizeof(payload[0])));
  1690. }
  1691. }