2

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

工具

我使用：Android Studio 3.0.1

构建 #AI-171.4443003，构建于 2017 年 11 月 9 日

JRE：1.8.0_152-release-915-b01 amd64

JVM：JetBrains sro 的 OpenJDK 64 位服务器 VM

Linux 4.15.0-29-通用

安卓-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。

4

1 回答 1

-1

尝试在您的机器中构建 Caffe2。AICamera 项目的 Caffe2 二进制文件对于新训练的模型来说有点陈旧。我也遇到了类似的问题，但现在我有了自定义模型的工作代码。

快乐编码:)

于 2018-08-30T08:15:50.420 回答