有两个输入的网络 ，量化时--source-file source.txt 如何设置呢？

VIM3
23

@liushuan 你的SDK版本为什么是6.4.0.10,你应该用最新你的SDK

24

@Frank
我现在使用了最新的版本，还是报了下面的错误
./…/khadas/aml_npu_sdk_6.4.3/acuity-toolkit/bin/pegasus generate inputmeta --model detection.json --separated-database True
usage: pegasus [-h]
{import,export,generate,prune,inference,quantize,train,dump,help}
…
pegasus: error: unrecognized arguments: --separated-database True

25

这个SDK需要更新下，我们下周会出一个新的版本。等等吧。

26

@numbqq
嗯嗯，可以考虑 输入不只是图片 是一个 tensor 的时候 如何处理，目前我的模型有两个输入，一个是 图片 287 287 3 另外一个是 tensor 维度是 6 6 256 这种如何处理

27

@liushuan 链接: https://pan.baidu.com/s/18U0wQ7-BZpd5r4A4zAHhHA 密码: n65o

这是新的SDK,你可以试一下,

28

我用了你新给的模型，第一步就报错了。
第一步命令：…/…/khadas/aml_npu_sdk/acuity-toolkit/bin/pegasus import pytorch --model detection2.pt --input-size-list “3,287,287#3,127,127”

Traceback (most recent call last):
File “pegasus.py”, line 131, in
File “pegasus.py”, line 112, in main
File “acuitylib/app/importer/commands.py”, line 282, in execute
File “acuitylib/vsi_nn.py”, line 213, in load_pytorch_by_onnx_backend
File “acuitylib/vsi_nn.py”, line 135, in load_onnx
File “acuitylib/app/importer/import_onnx.py”, line 116, in run
File “acuitylib/converter/convert_onnx.py”, line 62, in init
File “acuitylib/converter/convert_onnx.py”, line 321, in _shape_inference
File “acuitylib/onnx_ir/onnx_numpy_backend/shape_inference.py”, line 64, in infer_shape
AttributeError: ‘list’ object has no attribute ‘dtype’
[3814] Failed to execute script pegasus

29

链接: https://pan.baidu.com/s/1_ydf5EfFWF2k5Hb2xRRkAA 提取码: 6gu2 复制这段内容后打开百度网盘手机App，操作更方便哦
这里面是我的模型，有两个输入， 输入是 1, 3, 287, 287 # 1, 3, 127, 127 你能尝试看下转的流程能走通吗？我这边转换不通呢。

30

@numbqq @Frank
Siamese PRN 这个网络是有两个输入的，能不能帮忙看下模型能不能转到你们的NPU上。

31

image
image1702×698 198 KB
网络是这样的

32

@liushuan 模型转换工具是支持多输入模型的,你可以查看转换文档,按照文档转换看看,之前也有人实现了多输入模型

33

我尝试了好多次都没有成功，你说的成功的那个是基于什么的模型呢？（onnx , caffe, pytorch, tensorflow）

34

@liushuan 这个用户没有告诉我们,是他自己训练的模型

35

@Frank 还有一个问题，如果我的模型输入不是 图片 而是一个 特征图，例如 特征维度为 【20，10，10】 该如何修改 source_file 呢？ 用npy? --source text 又该如何修改呢？