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

[BUG] slice forward crash #112

Closed
huanyingjun opened this issue Aug 26, 2017 · 3 comments
Closed

[BUG] slice forward crash #112

huanyingjun opened this issue Aug 26, 2017 · 3 comments

Comments

@huanyingjun
Copy link

slice.cpp forward 函数中
我的proto中的参数如下
Slice slice2 1 3 to_use1 not_use2 to_use2 not_use22 3 2 8 -233
如下函数:

for (size_t i=0; i<top_blobs.size(); i++)
{
int slice = slices_ptr[i];
if (slice == -233)
{
slice = (channels - q) / (top_blobs.size() - i);
}

当输入的bottom 的 w = 11, h = 11, c =8, top_blobs.size() = 3
当 i = 2时, q=10
slice = (channels - q) / (top_blobs.size() - i);
slice = -2 ,这个明显不对
下面的 top_blob.create(w, h, slice); crash

@nihui
Copy link
Member

nihui commented Aug 26, 2017

slice 的参数是 num_slice 后面接着 num_slice 个数字表示输出的各个 c,最后几个可以 -233 代表平均切分
bottom c=8,slice 的三个输出 c 为 2 + 8 + x,已经超出输入c大小了,所以会 crash

@cutejerry
Copy link

slice层应该可以根据axis选择从N x C x H x W这0~3个维度中的某一个进行切分吧?目前ncnn好像只支持了按照Channel进行切分,可以加上其它维度吗?

@huanyingjun
Copy link
Author

是我搞错了,我的caffe模型中axis=2,是对H 进行切分,而ncnn 默认只支持对channel进行切分,所以才会crash。

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