关于MXNet量化的问题

[复制链接] · 发表于 2020-5-28 16:44:11

有两个问题希望帮忙解决一下：
1、在跑\rknn-toolkit-V1.3.0\examples\mxnet\fcn_resnet101，里面的测试案例，没有找到相应的（.json和.param）模型文件
2、在运行我自己的mxnet文件报错：
Parsing layer
E op _minus_scalar was not implemented
I ----------------Warning(0)----------------
E Catch exception when loading mxnet model: E:/project/Python/01RKNN/Face_model_MxNet/model-symbol.json!
E Traceback (most recent call last):
E File "rknn\api\rknn_base.py", line 588, in rknn.api.rknn_base.RKNNBase.load_mxnet
E File "rknn\base\RKNNlib\app\importer\mxnetloader.py", line 554, in rknn.base.RKNNlib.app.importer.mxnetloader.MxnetLoader.load
E File "rknn\base\RKNNlib\app\importer\mxnetloader.py", line 416, in rknn.base.RKNNlib.app.importer.mxnetloader.MxnetLoader.parse_layers
E File "rknn\base\RKNNlib\RKNNlog.py", line 327, in rknn.base.RKNNlib.RKNNlog.RKNNLog.e
E ValueError: op _minus_scalar was not implemented
Load model failed!

只看该作者 · 发表于 2020-5-30 17:53:49

本帖最后由 jefferyzhang 于 2020-5-30 17:55 编辑

这里的意思是 _minus_scalar 这个op不支持。
我可以帮你把这个op包给NPU部门，等他们支持，
但是mxnet用的客户不太多，你可以尝试下把mxnet转onnx再转rknn试试，或者转其他例如tf、caffe、pytorch这些比较热门的模型框架支持会更多一点。

如果需要我上报这个不支持的OP，需要提供：1. 你的mxnet原始模型
2. 你写的转换脚本
3. mxnet版本号
4. rknn-toolkit版本号（如果不是1.3.2请在1.3.2上试验后再发给我）