Skip to content

Commit fe10e50

Browse files
authored
Merge pull request #3821 from asmorkalov:as/new_dnn_caffe
Do not use forward to layer feature in WeChat to support the new dnn engine
2 parents bd3b734 + aaad40a commit fe10e50

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

modules/wechat_qrcode/src/detector/ssd_detector.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ vector<Mat> SSDDetector::forward(Mat img, const int target_width, const int targ
2424
false, false);
2525
net_.setInput(input, "data");
2626

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

0 commit comments

Comments
 (0)