DayBreak-u / chineseocr_lite

超轻量级中文ocr,支持竖排文字识别, 支持ncnn、mnn、tnn推理 ( dbnet(1.8M) + crnn(2.5M) + anglenet(378KB)) 总模型仅4.7M
GNU General Public License v2.0
11.83k stars 2.26k forks source link

C++ Ncnn版本的库,setGpuIndex后,退出进程时会崩溃 #372

Open MistEO opened 3 years ago

MistEO commented 3 years ago
#include <iostream>
#include "OcrLiteCaller.h"

int main(int argc, char **argv) 
{
    {
        OcrLiteCaller ocr;
        ocr.setGpuIndex(0);
    }
    std::cout << "end" << std::endl;    // Already print out
    return 0;
}

image

不知道是不是我之前封装的问题?但感觉不像呀,毕竟也没设置其他东西,我试了detect一些图片,都正常,但是进程退出的时候就会崩溃,我用另一台有独显的电脑测试同样会崩溃
另外我还尝试了用纯C的接口__stdcall重新封装了一遍,同样有这个崩溃问题
但如果我setGpuIndex(-1)则没有这个崩溃问题,一切正常;在ONNX的封装上也没有这个崩溃问题