1、RK3399Pro MNIST RKNN量化教程
一.搭建网络
model.py,同tensorflow官网mnist的例子差不多，不懂的可以去官网看下官网代码解析
二.训练网络
train.py,这里代码兼容了tf伪量化的代码，这里我们把create_training_graph()传入的参数is_quantify设为False就可以了，由于mnist拿到的train和test数据shape都是（784，),这里定义了一个reshape_batch函数把train时的batch以及test时的输入都reshape成（28,28），具体代码如下：
原作者：peng
更多回帖
