proc_lib.cc 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. // Copyright (c) 2022 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. #include "ultra_infer/vision/common/processors/proc_lib.h"
  15. namespace ultra_infer {
  16. namespace vision {
  17. ProcLib DefaultProcLib::default_lib = ProcLib::DEFAULT;
  18. std::ostream &operator<<(std::ostream &out, const ProcLib &p) {
  19. switch (p) {
  20. case ProcLib::DEFAULT:
  21. out << "ProcLib::DEFAULT";
  22. break;
  23. case ProcLib::OPENCV:
  24. out << "ProcLib::OPENCV";
  25. break;
  26. case ProcLib::FLYCV:
  27. out << "ProcLib::FLYCV";
  28. break;
  29. case ProcLib::CUDA:
  30. out << "ProcLib::CUDA";
  31. break;
  32. case ProcLib::CVCUDA:
  33. out << "ProcLib::CVCUDA";
  34. break;
  35. default:
  36. FDASSERT(false, "Unknown type of ProcLib.");
  37. }
  38. return out;
  39. }
  40. } // namespace vision
  41. } // namespace ultra_infer