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