Torchvision.ops.nms on Android Mobile

Ludwig_Friborg · May 28, 2021, 12:22pm

I was able to successfully use the operations by loading them with facebooks native loading. Here’s some code to load torchvision_ops:

import com.facebook.soloader.nativeloader.NativeLoader;
import com.facebook.soloader.nativeloader.SystemDelegate;

With this in main:

static {    
    if (!NativeLoader.isInitialized()) {
        NativeLoader.init(new SystemDelegate());
    }
    NativeLoader.loadLibrary("pytorch_jni");
    NativeLoader.loadLibrary("torchvision_ops");
}