Hi3516(海思)训练yolov5-6.0-->转oonx-->转caffe-转.wk文件
2:nnie的reshape也只支持4维,且第一维必须是0,为了能用nnie的reshape,我们不得不把x和y共享一个维度,这导致的结果是输出结果中,x和y在同一行,我们只需按个数取值即可。1:nnie不支持5个维度的permute(即transpose),且只支持0231的方式,过于局限,我们不妨删掉这一层,在后处理中按照合适的读取方式去找结果就好了。2:将view原来的输出维度(bs, na, no, ny, nx) 改为 (bs, na, no, ny * nx);1:去掉了原先的permute;
原创
2023-06-14 11:01:23 ·
1968 阅读 ·
7 评论