YOUNGLirr的博客

1、该文件可实现lstm单batch的多TimeStep循环运行，供大家学习。 2、其中，权重是从onnx中读出来保存为txt。由于onnx和pth文件吧Wi、Wf、Wo、Wg合在一个矩阵中，在pth中的顺序为i、f、g、o。在onnx中的顺序是i、o、f、g。模型的输入有三个分别为weight_ini是权重矩阵，input是输入矩阵，result是结果保存的地方，Ct_0,ht_0是LSTM的两个状态，可以初始化为0，初始化函数已经写好名为initializeArrays，这个初始化函数还可以初始化input

本程序是把“cccandppp”人博客中的基于opencv+opencvSVM改造而成的，我并没有用opencv里的mlp训练，因为不方便查看，也不方便学习，我用pytorch改写的，可查看实时训练结果。虽然准确率低，但是基本学习还是够用的，后期只要把MLP换成其他高级算法就可以了。pytorch框架非常便于改写里面的算法。