update model_process.cpp

This commit is contained in:
chen68
2020-10-19 01:50:36 -07:00
parent bf21c4c872
commit 22b83024f5
+23 -23
View File
@@ -443,7 +443,7 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
int amount_onebatch = 1; int amount_onebatch = 1;
for (int j = 1; j < dim->dimCount; j++) { for (int j = 1; j < dim->dimCount; j++) {
amount_onebatch *= dim->dims[j]; amount_onebatch *= dim->dims[j];
} }
switch (datatype) { switch (datatype) {
case 0: case 0:
for (int i = 0; i < len / sizeof(float); i++) { for (int i = 0; i < len / sizeof(float); i++) {
@@ -452,10 +452,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){ if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n"; outstr << "\n\n";
} else{ } else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){ if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
} }
break; break;
case 1: case 1:
@@ -470,7 +470,7 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
} }
} }
} }
break; break;
case 2: case 2:
for (int i = 0; i < len / sizeof(int8_t); i++) { for (int i = 0; i < len / sizeof(int8_t); i++) {
@@ -483,7 +483,7 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << "\n"; outstr << "\n";
} }
} }
} }
break; break;
case 3: case 3:
for (int i = 0; i < len / sizeof(int); i++) { for (int i = 0; i < len / sizeof(int); i++) {
@@ -492,10 +492,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){ if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n"; outstr << "\n\n";
} else{ } else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){ if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
} }
break; break;
case 4: case 4:
@@ -505,10 +505,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){ if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n"; outstr << "\n\n";
} else{ } else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){ if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
} }
break; break;
case 6: case 6:
@@ -518,10 +518,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){ if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n"; outstr << "\n\n";
} else{ } else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){ if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
} }
break; break;
case 7: case 7:
@@ -531,10 +531,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){ if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n"; outstr << "\n\n";
} else{ } else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){ if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
} }
break; break;
case 8: case 8:
@@ -544,10 +544,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){ if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n"; outstr << "\n\n";
} else{ } else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){ if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
} }
break; break;
case 9: case 9:
@@ -557,10 +557,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){ if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n"; outstr << "\n\n";
} else{ } else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){ if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
} }
break; break;
case 10: case 10:
@@ -570,10 +570,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){ if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n"; outstr << "\n\n";
} else{ } else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){ if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
} }
break; break;
case 11: case 11:
@@ -583,10 +583,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){ if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n"; outstr << "\n\n";
} else{ } else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){ if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
} }
break; break;
case 12: case 12:
@@ -596,10 +596,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){ if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n"; outstr << "\n\n";
} else{ } else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){ if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n"; outstr << "\n";
} }
} }
} }
break; break;
default: default: