mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-06 06:26:29 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #19035 from berak:fix_dnn_net_dump_colors

Browse Source
This commit is contained in:
Alexander Alekhin 2020-12-07 22:11:49 +00:00
commit 40ca8f4695
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/dnn/src/dnn.cpp
View File

@ -4461,7 +4461,7 @@ string Net::Impl::dump()
prevNode = itBackend->second; prevNode = itBackend->second;
} }
} }
string colors[] = {"#ffffb3", "#fccde5", "#8dd3c7", "#bebada", "#80b1d3", "#fdb462", "#ff4848", "#b35151"}; std::vector<string> colors = {"#ffffb3", "#fccde5", "#8dd3c7", "#bebada", "#80b1d3", "#fdb462", "#ff4848", "#b35151", "#b266ff"};
string backend; string backend;
switch (prefBackend) switch (prefBackend)
{ {
@ -4613,6 +4613,7 @@ string Net::Impl::dump()
case DNN_TARGET_CUDA_FP16: out << "CUDA_FP16"; colorId = 6; break; case DNN_TARGET_CUDA_FP16: out << "CUDA_FP16"; colorId = 6; break;
// don't use default: // don't use default:
} }
CV_Assert(colorId < colors.size());
out << "\\n"; // align center out << "\\n"; // align center
out << ((clusterIds.size() == 1)? "\" " : " }\" "); out << ((clusterIds.size() == 1)? "\" " : " }\" ");
out << "fillcolor=\"" << colors[colorId] << "\" "; out << "fillcolor=\"" << colors[colorId] << "\" ";