RK3399Pro入门教程（9）MNIST RKNN量化教程

显示全部楼层 · 发表于 2019-9-21 14:50:47

本帖最后由 puyanan 于 2019-9-21 14:52 编辑

楼主，你好，这个例子里测试模型没有dropout层，输入只有一个，正好rknn只支持单输入。如果一个网络里的每一个conv2D都有BN层，最后还有Dropout层。tensorflow的pb文件中输入有两个，“input”和“phase_train”，其中phase_train节点输出连接每一个conv2D，怎么把这种情况改成“单输入”呢？修改模型，将phase_train固定为Flase再重新固化成pb模型，请问用什么工具实现呢？