Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

cnnctc solve core dumped problem

This commit is contained in:
chenweitao_295 2021-06-26 18:14:37 +08:00
parent ac6d75b803
commit 803ff2d326
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
model_zoo/official/cv/cnnctc/ascend310_infer/src/main.cc
View File

@ -140,19 +140,17 @@ int main(int argc, char **argv) {
composeTranspose(imgPad, &img);
if (NewWidth < FLAGS_image_width) {
int img_size = FLAGS_image_width * FLAGS_image_height * 3;
void *netOutput;
float *address;
netOutput = img.MutableData();
address = static_cast<float *>(netOutput);
for (int j = 0; j <= img_size; j += FLAGS_image_width) {
float *netOutput;
netOutput = static_cast<float *>(img.MutableData());
for (int j = 0; j < img_size; j += FLAGS_image_width) {
float temp = 0;
address = address + NewWidth - 1;
temp = *address;
netOutput = netOutput + NewWidth - 1;
temp = *netOutput;
int n = NewWidth;
address++;
netOutput++;
while (n < FLAGS_image_width) {
*address = temp;
address++;
*netOutput = temp;
netOutput++;
n++;
}
}