一、前言
最近项目因一些原因停止,空下来的我给自己找点事做,所以就把眼光瞄上了行人检测,该模型是我一个目标检测群群主发的,他在知乎和git都有账号,不过我忘了。。。不说其他了,开始吧。demo地址:demo地址
二、模型的选择与转换
模型是一切的基础,之所以选定这个模型也是因为它的op海思都支持,没什么很新的op,使得转换模型成为可能。模型平面图就不看了,太大了,我等下提供gitee仓库的地址,有需要自己下去看吧。
1.模型转换
原模型是pytorch版本,所以先要转化成onnx,再转化成caffe,最后转化成nnie-wk格式,在这里费了点时间,时间过去有点久了,我尽量回忆下(抱歉抱歉)。
转换路线为 pytorch->onnx->caffe, 其中onnx->caffe demo: https://gitee.com/huiwei13/onnx2caffe?_from=gitee_search ,这样的demo网上基本都差不多,不知道初始版本是谁写的。我pytorch版本是1.5的,网上也有pytorch直接转caffe的demo,但是都要求0.4及以下的pytorch版本,不好下而且这样的版本打开pytorch1.5的模型也容易报错,此外pytorch与caffe两个框架对于maxpool的实现有所不同,即使转换成功,可能最后模型还是不能用。所以还是建议用onnx做中转