参考:https://blog.csdn.net/tangshopping/article/details/110038605
总共有三个地方需要修改。
1.focus层:先slice再conv。这个博客中给出三个思路,实现了一种方法。但是看评论区说不稳定和相似度偏低。没有尝试,先用不带fouces层(直接用conv代替)的模型。
2.transpose层,yolov5的transpose层,即permute是5个维度的。但是海思只支持(0,1,2,3)->(0,2,3,1)的转化。
博客中给出的方法是直接删除:
3.reshape层。yolov5中的reshape层是5个维度的。但是nnie只支持C、H、W的。博客中给出的方法:
改成下面这样: