rknn-toolkit支持pytorch1.4.0下efficientDetD1的模型为rknn模型吗

[复制链接] · 发表于 2020-8-10 16:35:24

本帖最后由 daigang 于 2020-8-10 16:37 编辑

rknn-toolkit目前支持pytorch1.4.0吗？能否成功转换pytorch1.4.0下的efficientDetD1的模型为rknn模型？使用的是https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch的代码进行模型训练和保存的模型，但是不知道如何才能成功转换为rknn模型，能否提供点建议。

显示全部楼层 · 发表于 2020-8-17 14:42:22

jefferyzhang 发表于 2020-8-10 16:55
我们做芯片角度只考虑op是不是支持，不用问什么什么模型支不支持，模型每年增长上千个，我们怎么可能一个个 ...

在网上找到的tensorflow下的efficientDet的代码都是tensorflow2.0下的代码，之前看贵公司说rknn-toolkit-1.3.2目前还不支持tensorflow2.0，就没有使用了。目前训练的一版是pytorch1.4.0下efficientDetD1的模型，也是遇到了转换模型的问题。