Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

关于代码 #5

Open
jiejiangd opened this issue Apr 26, 2023 · 9 comments
Open

关于代码 #5

jiejiangd opened this issue Apr 26, 2023 · 9 comments

Comments

@jiejiangd
Copy link

您好,您在readme里红绿框标注的部分是不是没有体现在第四部分的代码里面哦,这是两种方式吗,第四部分的代码好像跟那个什么shoppingtang差不多,主要是我跑他那个有点问题

@jiejiangd
Copy link
Author

我是按照您的第二部分的yolov5训练的,然后用4部署,得到好多乱七八糟的框,是什么问题哦

@liujianisme
Copy link
Owner

你好,我用的是sample_nnie_main.c里面的case '8'的接口,这个接口使用的是海思官方的源代码,代码导航的话可以看到相应的修改的

@jiejiangd
Copy link
Author

你好,我现在能跑通了,用的您的第二部分的yolov5训练的,我发现您用的yolov5s-6.0好像是有改过的,模型小了很多是吧

@liujianisme
Copy link
Owner

你好,原版的YOLO5我有过改动,主要是将原来的激活函数改成了Relu、upsample层改成了反卷积层、网络模型层的话我是有裁剪的

@jiejiangd
Copy link
Author

你好,我发现一个问题就是caffe转成wk之后无法识别小目标了,您有遇到类似的问题吗

@liujianisme
Copy link
Owner

我这边没有碰到过类似的问题,你可以比对下caffe和海思板子上的输出是否一致

@jiejiangd
Copy link
Author

你好,想问一下为什么我改了这两行之后速度巨慢啊
param.input_blob[i] = (hi_s32*)nnie_param->seg_data[0].dst[2-i].virt_addr;
param.input_stride[i] = nnie_param->seg_data[0].dst[2-i].stride;

@liujianisme
Copy link
Owner

这个不清楚,有可能是你的代码这里改动之后导致后面的框数量激增,框的数量增加之后会相应的导致后处理的时间会增大,这个需要你去调试一下看看

@gongzhenwei1
Copy link

你好,我用的是sample_nnie_main.c里面的case '8'的接口,这个接口使用的是海思官方的源代码,代码导航的话可以看到相应的修改的
您好 我看您给的sample_nnie_main.c里面有case'b‘接口,这个接口是可以直接拿来跑yolov5的吗?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants