From 9bb3d5ef77ba87b999141db9f1a5eed9e82ad745 Mon Sep 17 00:00:00 2001 From: wang-xinyu Date: Tue, 21 Jul 2020 10:50:36 +0800 Subject: [PATCH] fix large input segfault --- yolov3-spp/yolov3-spp.cpp | 2 +- yolov3-tiny/yolov3-tiny.cpp | 2 +- yolov3/yolov3.cpp | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/yolov3-spp/yolov3-spp.cpp b/yolov3-spp/yolov3-spp.cpp index 1bc0611..b1dd5e0 100644 --- a/yolov3-spp/yolov3-spp.cpp +++ b/yolov3-spp/yolov3-spp.cpp @@ -516,7 +516,7 @@ int main(int argc, char** argv) { } // prepare input data --------------------------- - float data[3 * INPUT_H * INPUT_W]; + static float data[3 * INPUT_H * INPUT_W]; //for (int i = 0; i < 3 * INPUT_H * INPUT_W; i++) // data[i] = 1.0; static float prob[OUTPUT_SIZE]; diff --git a/yolov3-tiny/yolov3-tiny.cpp b/yolov3-tiny/yolov3-tiny.cpp index beead67..404d9f6 100644 --- a/yolov3-tiny/yolov3-tiny.cpp +++ b/yolov3-tiny/yolov3-tiny.cpp @@ -420,7 +420,7 @@ int main(int argc, char** argv) { } // prepare input data --------------------------- - float data[3 * INPUT_H * INPUT_W]; + static float data[3 * INPUT_H * INPUT_W]; //for (int i = 0; i < 3 * INPUT_H * INPUT_W; i++) // data[i] = 1.0; static float prob[OUTPUT_SIZE]; diff --git a/yolov3/yolov3.cpp b/yolov3/yolov3.cpp index 64d1439..26378cc 100644 --- a/yolov3/yolov3.cpp +++ b/yolov3/yolov3.cpp @@ -505,7 +505,7 @@ int main(int argc, char** argv) { } // prepare input data --------------------------- - float data[3 * INPUT_H * INPUT_W]; + static float data[3 * INPUT_H * INPUT_W]; //for (int i = 0; i < 3 * INPUT_H * INPUT_W; i++) // data[i] = 1.0; static float prob[OUTPUT_SIZE];