vision.h 4.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  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. #pragma once
  15. #include "ultra_infer/core/config.h"
  16. #ifdef ENABLE_VISION
  17. #include "ultra_infer/vision/classification/contrib/resnet.h"
  18. #include "ultra_infer/vision/classification/contrib/yolov5cls/yolov5cls.h"
  19. #include "ultra_infer/vision/classification/ppcls/model.h"
  20. #include "ultra_infer/vision/classification/ppshitu/ppshituv2_det.h"
  21. #include "ultra_infer/vision/classification/ppshitu/ppshituv2_rec.h"
  22. #include "ultra_infer/vision/detection/contrib/fastestdet/fastestdet.h"
  23. #include "ultra_infer/vision/detection/contrib/nanodet_plus.h"
  24. #include "ultra_infer/vision/detection/contrib/rknpu2/model.h"
  25. #include "ultra_infer/vision/detection/contrib/scaledyolov4.h"
  26. #include "ultra_infer/vision/detection/contrib/yolor.h"
  27. #include "ultra_infer/vision/detection/contrib/yolov5/yolov5.h"
  28. #include "ultra_infer/vision/detection/contrib/yolov5lite.h"
  29. #include "ultra_infer/vision/detection/contrib/yolov5seg/yolov5seg.h"
  30. #include "ultra_infer/vision/detection/contrib/yolov6.h"
  31. #include "ultra_infer/vision/detection/contrib/yolov7/yolov7.h"
  32. #include "ultra_infer/vision/detection/contrib/yolov7end2end_ort.h"
  33. #include "ultra_infer/vision/detection/contrib/yolov7end2end_trt.h"
  34. #include "ultra_infer/vision/detection/contrib/yolov8/yolov8.h"
  35. #include "ultra_infer/vision/detection/contrib/yolox.h"
  36. #include "ultra_infer/vision/detection/ppdet/model.h"
  37. #include "ultra_infer/vision/facealign/contrib/face_landmark_1000.h"
  38. #include "ultra_infer/vision/facealign/contrib/pfld.h"
  39. #include "ultra_infer/vision/facealign/contrib/pipnet.h"
  40. #include "ultra_infer/vision/facedet/contrib/centerface/centerface.h"
  41. #include "ultra_infer/vision/facedet/contrib/retinaface.h"
  42. #include "ultra_infer/vision/facedet/contrib/scrfd.h"
  43. #include "ultra_infer/vision/facedet/contrib/ultraface.h"
  44. #include "ultra_infer/vision/facedet/contrib/yolov5face.h"
  45. #include "ultra_infer/vision/facedet/contrib/yolov7face/yolov7face.h"
  46. #include "ultra_infer/vision/facedet/ppdet/blazeface/blazeface.h"
  47. #include "ultra_infer/vision/faceid/contrib/adaface/adaface.h"
  48. #include "ultra_infer/vision/faceid/contrib/insightface/model.h"
  49. #include "ultra_infer/vision/generation/contrib/animegan.h"
  50. #include "ultra_infer/vision/headpose/contrib/fsanet.h"
  51. #include "ultra_infer/vision/keypointdet/pptinypose/pptinypose.h"
  52. #include "ultra_infer/vision/matting/contrib/modnet.h"
  53. #include "ultra_infer/vision/matting/contrib/rvm.h"
  54. #include "ultra_infer/vision/matting/ppmatting/ppmatting.h"
  55. #include "ultra_infer/vision/ocr/ppocr/classifier.h"
  56. #include "ultra_infer/vision/ocr/ppocr/dbcurvedetector.h"
  57. #include "ultra_infer/vision/ocr/ppocr/dbdetector.h"
  58. #include "ultra_infer/vision/ocr/ppocr/ppocr_v2.h"
  59. #include "ultra_infer/vision/ocr/ppocr/ppocr_v3.h"
  60. #include "ultra_infer/vision/ocr/ppocr/ppocr_v4.h"
  61. #include "ultra_infer/vision/ocr/ppocr/ppstructurev2_layout.h"
  62. #include "ultra_infer/vision/ocr/ppocr/ppstructurev2_table.h"
  63. #include "ultra_infer/vision/ocr/ppocr/recognizer.h"
  64. #include "ultra_infer/vision/ocr/ppocr/structurev2_layout.h"
  65. #include "ultra_infer/vision/ocr/ppocr/structurev2_ser_vi_layoutxlm.h"
  66. #include "ultra_infer/vision/ocr/ppocr/structurev2_table.h"
  67. #include "ultra_infer/vision/ocr/ppocr/utils/ocr_utils.h"
  68. #include "ultra_infer/vision/ocr/ppocr/uvdocwarpper.h"
  69. #include "ultra_infer/vision/perception/paddle3d/caddn/caddn.h"
  70. #include "ultra_infer/vision/perception/paddle3d/centerpoint/centerpoint.h"
  71. #include "ultra_infer/vision/perception/paddle3d/petr/petr.h"
  72. #include "ultra_infer/vision/perception/paddle3d/smoke/smoke.h"
  73. #include "ultra_infer/vision/segmentation/ppseg/model.h"
  74. #include "ultra_infer/vision/sr/ppsr/model.h"
  75. #include "ultra_infer/vision/tracking/pptracking/model.h"
  76. #endif
  77. #include "ultra_infer/vision/visualize/visualize.h"