We are trying to generate a shared object file to implement alphapose on android. We ran the following code from neo-ai-dlr:
#!/usr/bin/env python3
from tvm import relay
from gluoncv.model_zoo import get_model
from tvm_compiler_utils import tvm_compile
model_name, dlr_model_name = "alpha_pose_resnet101_v1b_coco", "dlr_gluoncv_alpha_pose_resnet101_v1b_coco"
#model_name, dlr_model_name = "mobilenetv2_1.0", "dlr_gluoncv_mobilenet_v2_100"
#model_name, dlr_model_name = "resnet18_v2", "dlr_gluoncv_resnet18_v2"
#model_name, dlr_model_name = "resnet50_v2", "dlr_gluoncv_resnet50_v2"
shape_dict = {'data': (1, 3, 800, 800)}
dtype='float32'
print("Model:", model_name, ", shape_dict:", shape_dict)
block = get_model(model_name, pretrained=True)
for arch in ["arm64-v8a", "armeabi-v7a", "x86_64", "x86"]:
sym, params = relay.frontend.from_mxnet(block, shape=shape_dict, dtype=dtype)
func = sym["main"]
func = relay.Function(func.params, relay.nn.softmax(func.body), None, func.type_params, func.attrs)
tvm_compile(func, params, arch, dlr_model_name)`
Full error traceback:
sys.getrecursionlimit() 1031
Model: alpha_pose_resnet101_v1b_coco , shape_dict: {'data': (1, 3, 800, 800)}
Traceback (most recent call last):
File "compile_gluoncv.py", line 18, in
sym, params = relay.frontend.from_mxnet(block, shape=shape_dict, dtype=dtype)
File "/home/bemrr/anaconda3/envs/neodlr/lib/python3.6/site-packages/tvm/relay/frontend/mxnet.py", line 1197, in from_mxnet
func = _from_mxnet_impl(sym, shape, dtype, mod)
File "/home/bemrr/anaconda3/envs/neodlr/lib/python3.6/site-packages/tvm/relay/frontend/mxnet.py", line 1112, in _from_mxnet_impl
'Operator {} is not supported in frontend MXNet.'.format(op_name))
tvm.error.OpNotImplemented: Operator broadcast_like is not supported in frontend MXNet.
Please guide how to solve this error!