Skip to content

Commit

Permalink
Merge pull request #3821 from asmorkalov:as/new_dnn_caffe
Browse files Browse the repository at this point in the history
Do not use forward to layer feature in WeChat to support the new dnn engine
  • Loading branch information
asmorkalov authored Nov 6, 2024
2 parents bd3b734 + aaad40a commit fe10e50
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions modules/wechat_qrcode/src/detector/ssd_detector.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ vector<Mat> SSDDetector::forward(Mat img, const int target_width, const int targ
false, false);
net_.setInput(input, "data");

auto prob = net_.forward("detection_output");
auto prob = net_.forward();
vector<Mat> point_list;
// the shape is (1,1,100,7)=>(batch,channel,count,dim)
for (int row = 0; row < prob.size[2]; row++) {
Expand Down Expand Up @@ -54,4 +54,4 @@ vector<Mat> SSDDetector::forward(Mat img, const int target_width, const int targ
return point_list;
}
} // namespace wechat_qrcode
} // namespace cv
} // namespace cv

0 comments on commit fe10e50

Please sign in to comment.