utility.h 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. // Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. #pragma once
  15. #include <chrono>
  16. #include <iomanip>
  17. #include <iostream>
  18. #include <ostream>
  19. #include <stdlib.h>
  20. #include <vector>
  21. #include <algorithm>
  22. #include <cstring>
  23. #include <fstream>
  24. #include <numeric>
  25. #include "opencv2/core.hpp"
  26. #include "opencv2/imgcodecs.hpp"
  27. #include "opencv2/imgproc.hpp"
  28. namespace PaddleOCR {
  29. class Utility {
  30. public:
  31. static std::vector<std::string> ReadDict(const std::string &path);
  32. static void
  33. VisualizeBboxes(const cv::Mat &srcimg,
  34. const std::vector<std::vector<std::vector<int>>> &boxes);
  35. template <class ForwardIterator>
  36. inline static size_t argmax(ForwardIterator first, ForwardIterator last) {
  37. return std::distance(first, std::max_element(first, last));
  38. }
  39. };
  40. } // namespace PaddleOCR