You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

Pipeline.h 1.1 kB

6 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. //
  2. // Created by 庾金科 on 22/10/2017.
  3. //
  4. #ifndef SWIFTPR_PIPLINE_H
  5. #define SWIFTPR_PIPLINE_H
  6. #include "PlateDetection.h"
  7. #include "PlateSegmentation.h"
  8. #include "CNNRecognizer.h"
  9. #include "PlateInfo.h"
  10. #include "FastDeskew.h"
  11. #include "FineMapping.h"
  12. #include "Recognizer.h"
  13. namespace pr{
  14. class PipelinePR{
  15. public:
  16. GeneralRecognizer *generalRecognizer;
  17. PlateDetection *plateDetection;
  18. PlateSegmentation *plateSegmentation;
  19. FineMapping *fineMapping;
  20. PipelinePR(std::string detector_filename,
  21. std::string finemapping_prototxt,std::string finemapping_caffemodel,
  22. std::string segmentation_prototxt,std::string segmentation_caffemodel,
  23. std::string charRecognization_proto,std::string charRecognization_caffemodel
  24. );
  25. ~PipelinePR();
  26. std::vector<std::string> plateRes;
  27. std::vector<PlateInfo> RunPiplineAsImage(cv::Mat plateImage);
  28. };
  29. }
  30. #endif //SWIFTPR_PIPLINE_H