android - 我们实例化Caffe2预测器时应用程序卡住了

当我们实例化Caffe2预测器时，应用程序卡住了!

工具类

我用:
Android Studio 3.0.1

Build＃AI-171.4443003，建于2017年11月9日

JRE:1.8.0_152-release-915-b01 amd64

JVM:JetBrains s.r.o的OpenJDK 64位服务器VM

Linux 4.15.0-29-通用

android-ndk-r13b

C++中的代码

#define alog(...) __android_log_print(ANDROID_LOG_ERROR, "F8DEMO", __VA_ARGS__);
void loadToNetDef(AAssetManager* mgr, caffe2::NetDef* net, const char *filename) {
    AAsset* asset = AAssetManager_open(mgr, filename, AASSET_MODE_BUFFER);
    assert(asset != nullptr);
    const void *data = AAsset_getBuffer(asset);
    assert(data != nullptr);
    off_t len = AAsset_getLength(asset);
    assert(len != 0);
    if (!net->ParseFromArray(data, len)) {
        alog("Couldn't parse net from data.\n");
    }
    AAsset_close(asset);
}
void Java_initCaffe2(JNIEnv* env, jobject /* this */, jobject assetManager){
    AAssetManager *mgr = AAssetManager_fromJava(env, assetManager);
    alog("Attempting to load protobuf netdefs...");
    loadToNetDef(mgr, &_initNet,   "init_net.pb");
    loadToNetDef(mgr, &_predictNet,"predict_net.pb");
    alog("done.");
    alog("Instantiating predictor...");
    _predictor = new caffe2::Predictor(_initNet, _predictNet);
    alog("done.")
}

追溯

08-07 17:10:00.308 897-897/com.example.augustinas.nc2 E/Gideon: Loaded module
08-07 17:10:00.315 897-1110/com.example.augustinas.nc2 E/F8DEMO: Attempting to load protobuf netdefs...
08-07 17:10:07.198 897-1110/com.example.augustinas.nc2 E/F8DEMO: Couldn't parse net from data.
08-07 17:10:07.287 897-1110/com.example.augustinas.nc2 E/F8DEMO: done.
08-07 17:10:07.287 897-1110/com.example.augustinas.nc2 E/F8DEMO: Instantiating predictor...
08-07 17:10:09.251 897-1110/com.example.augustinas.nc2 A/native: [F given_tensor_fill_op.h:27] Check failed: output->size() == values_.size() output size: 22937600 given size: 5348403
08-07 17:10:09.251 897-1110/com.example.augustinas.nc2 A/native: terminating.
08-07 17:10:09.251 897-1110/com.example.augustinas.nc2 A/libc: Fatal signal 6
(SIGABRT), code -6 in tid 1110 (AsyncTask #1)

原始资料:

我们使用此处显示的库和代码:https://github.com/caffe2/AICamera

使用型号

我们使用自定义的InceptionResNet模型。 init_net.pb和predict_net.pb的大小为155,1 MB和6.1 kB。

最佳答案

尝试在您的计算机中构建Caffe2。 AICamera Project的Caffe2 Binaries对于新训练的模型有些陈旧。我也遇到了类似的问题，但是现在有了自定义模型的工作代码。

快乐编码:)

关于android - 我们实例化Caffe2预测器时应用程序卡住了，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/51728496/