mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-13 13:13:26 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix Vulkan's max pooling in case of no output indices

Browse Source
This commit is contained in:
Dmitry Kurtaev 2018-11-15 14:10:24 +03:00
parent 8409aa9eba
commit ef5d921eac
1 changed files with 4 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
modules/dnn/src/vkcom/src/op_pool.cpp
View File

@ -90,16 +90,10 @@ bool OpPool::forward(std::vector<Tensor>& ins,
std::vector<Tensor>& blobs, std::vector<Tensor>& blobs,
std::vector<Tensor>& outs) std::vector<Tensor>& outs)
{ {
for (size_t ii = 0; ii < ins.size(); ii++) Tensor& inpMat = ins[0];
{ Tensor& outMat = outs[0];
Tensor& inpMat = ins[ii]; Tensor maskMat = outs.size() > 1 ? outs[1] : Tensor();
int out_index = (pool_type_ == kPoolTypeMax) ? 2 : 1; return forward(inpMat, outMat, maskMat);
Tensor& outMat = outs[out_index * ii];
Tensor maskMat = (pool_type_ == kPoolTypeMax) ? outs[2 * ii + 1] : Tensor();
if (!forward(inpMat, outMat, maskMat))
return false;
}
return true;
} }
bool OpPool::forward(Tensor& in, Tensor& out, Tensor& mask) bool OpPool::forward(Tensor& in, Tensor& out, Tensor& mask)