| @@ -1,5 +1,5 @@ | |||||
| // | // | ||||
| // Created by 庾金科 on 23/10/2017. | |||||
| // Created by Jack Yu on 23/10/2017. | |||||
| // | // | ||||
| #include "../include/Pipeline.h" | #include "../include/Pipeline.h" | ||||
| @@ -53,11 +53,8 @@ namespace pr { | |||||
| { | { | ||||
| image_finemapping = fineMapping->FineMappingHorizon(image_finemapping, 2, HorizontalPadding); | image_finemapping = fineMapping->FineMappingHorizon(image_finemapping, 2, HorizontalPadding); | ||||
| cv::resize(image_finemapping, image_finemapping, cv::Size(136+HorizontalPadding, 36)); | cv::resize(image_finemapping, image_finemapping, cv::Size(136+HorizontalPadding, 36)); | ||||
| // cv::imshow("image_finemapping",image_finemapping); | |||||
| // cv::waitKey(0); | |||||
| plateinfo.setPlateImage(image_finemapping); | plateinfo.setPlateImage(image_finemapping); | ||||
| std::vector<cv::Rect> rects; | std::vector<cv::Rect> rects; | ||||
| plateSegmentation->segmentPlatePipline(plateinfo, 1, rects); | plateSegmentation->segmentPlatePipline(plateinfo, 1, rects); | ||||
| plateSegmentation->ExtractRegions(plateinfo, rects); | plateSegmentation->ExtractRegions(plateinfo, rects); | ||||
| cv::copyMakeBorder(image_finemapping, image_finemapping, 0, 0, 0, 20, cv::BORDER_REPLICATE); | cv::copyMakeBorder(image_finemapping, image_finemapping, 0, 0, 0, 20, cv::BORDER_REPLICATE); | ||||
| @@ -69,33 +66,20 @@ namespace pr { | |||||
| //Segmentation-free | //Segmentation-free | ||||
| else if(method==SEGMENTATION_FREE_METHOD) | else if(method==SEGMENTATION_FREE_METHOD) | ||||
| { | { | ||||
| image_finemapping = fineMapping->FineMappingHorizon(image_finemapping, 4, HorizontalPadding+3); | image_finemapping = fineMapping->FineMappingHorizon(image_finemapping, 4, HorizontalPadding+3); | ||||
| cv::resize(image_finemapping, image_finemapping, cv::Size(136+HorizontalPadding, 36)); | cv::resize(image_finemapping, image_finemapping, cv::Size(136+HorizontalPadding, 36)); | ||||
| // cv::imwrite("./test.png",image_finemapping); | |||||
| // cv::imshow("image_finemapping",image_finemapping); | |||||
| // cv::waitKey(0); | |||||
| plateinfo.setPlateImage(image_finemapping); | plateinfo.setPlateImage(image_finemapping); | ||||
| // std::vector<cv::Rect> rects; | |||||
| std::pair<std::string,float> res = segmentationFreeRecognizer->SegmentationFreeForSinglePlate(plateinfo.getPlateImage(),pr::CH_PLATE_CODE); | std::pair<std::string,float> res = segmentationFreeRecognizer->SegmentationFreeForSinglePlate(plateinfo.getPlateImage(),pr::CH_PLATE_CODE); | ||||
| plateinfo.confidence = res.second; | plateinfo.confidence = res.second; | ||||
| plateinfo.setPlateName(res.first); | plateinfo.setPlateName(res.first); | ||||
| } | } | ||||
| results.push_back(plateinfo); | results.push_back(plateinfo); | ||||
| } | } | ||||
| // for (auto str:results) { | |||||
| // std::cout << str << std::endl; | |||||
| // } | |||||
| return results; | return results; | ||||
| }//namespace pr | }//namespace pr | ||||
| } | |||||
| } | |||||