diff --git a/modules/wechat_qrcode/src/detector/ssd_detector.cpp b/modules/wechat_qrcode/src/detector/ssd_detector.cpp index dca1851ae0d..52e8a635d56 100644 --- a/modules/wechat_qrcode/src/detector/ssd_detector.cpp +++ b/modules/wechat_qrcode/src/detector/ssd_detector.cpp @@ -24,7 +24,7 @@ vector 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 point_list; // the shape is (1,1,100,7)=>(batch,channel,count,dim) for (int row = 0; row < prob.size[2]; row++) { @@ -54,4 +54,4 @@ vector SSDDetector::forward(Mat img, const int target_width, const int targ return point_list; } } // namespace wechat_qrcode -} // namespace cv \ No newline at end of file +} // namespace cv