meta.py 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136
  1. # !/usr/bin/env python3
  2. # -*- coding: UTF-8 -*-
  3. ################################################################################
  4. #
  5. # Copyright (c) 2024 Baidu.com, Inc. All Rights Reserved
  6. #
  7. ################################################################################
  8. """
  9. Author: PaddlePaddle Authors
  10. """
  11. __all__ = ['get_all_repo_names']
  12. # REPO_NAMES = [
  13. # 'PaddleSeg', 'Paddle3D', 'PaddleClas', 'PaddleDetection', 'PaddleOCR',
  14. # 'PaddleTS', 'PaddleNLP', 'PaddleSpeech', 'PARL', 'PaddleMIX']
  15. REPO_NAMES = [
  16. 'PaddleClas', 'PaddleOCR', 'PaddleDetection', 'PaddleSeg', 'PaddleNLP',
  17. 'PaddleTS'
  18. ]
  19. REPO_META = {
  20. 'PaddleSeg': {
  21. 'repo_url': '/PaddlePaddle/PaddleSeg.git',
  22. 'branch': 'develop',
  23. 'pkg_name': 'paddleseg',
  24. 'lib_name': 'paddleseg',
  25. 'pdx_pkg_name': 'PaddleSeg_api',
  26. 'editable': False,
  27. 'extra_req_files': ['Matting/requirements.txt'],
  28. 'path_env': 'PADDLE_PDX_PADDLESEG_PATH',
  29. },
  30. 'Paddle3D': {
  31. 'repo_url': '/PaddlePaddle/Paddle3D.git',
  32. 'branch': 'develop',
  33. 'pkg_name': 'paddle3d',
  34. 'lib_name': 'paddle3d',
  35. 'pdx_pkg_name': 'Paddle3D_api',
  36. 'editable': False,
  37. 'path_env': 'PADDLE_PDX_PADDLE3D_PATH',
  38. 'requires': ['PaddleSeg', 'PaddleDetection'],
  39. 'pdx_pkg_deps': ['nuscenes-devkit', 'pyquaternion'],
  40. },
  41. 'PaddleClas': {
  42. 'repo_url': '/PaddlePaddle/PaddleClas.git',
  43. 'branch': 'develop',
  44. 'pkg_name': 'paddleclas',
  45. 'lib_name': 'paddleclas',
  46. 'pdx_pkg_name': 'PaddleClas_api',
  47. # PaddleClas must be installed in non-editable mode, otherwise it throws
  48. # an Import error.
  49. 'editable': False,
  50. 'path_env': 'PADDLE_PDX_PADDLECLAS_PATH',
  51. },
  52. 'PaddleDetection': {
  53. 'repo_url': '/PaddlePaddle/PaddleDetection.git',
  54. 'branch': 'develop',
  55. 'pkg_name': 'paddledet',
  56. 'lib_name': 'ppdet',
  57. 'pdx_pkg_name': 'PaddleDetection_api',
  58. 'editable': False,
  59. 'path_env': 'PADDLE_PDX_PADDLEDETECTION_PATH',
  60. },
  61. 'PaddleOCR': {
  62. 'repo_url': '/PaddlePaddle/PaddleOCR.git',
  63. 'platform': 'github',
  64. 'branch': 'release/2.6.1',
  65. 'pkg_name': 'paddleocr',
  66. 'lib_name': 'paddleocr',
  67. 'pdx_pkg_name': 'PaddleOCR_api',
  68. 'editable': False,
  69. 'extra_req_files': ['ppstructure/kie/requirements.txt'],
  70. 'path_env': 'PADDLE_PDX_PADDLEOCR_PATH',
  71. 'requires': ['PaddleNLP'],
  72. },
  73. 'PaddleTS': {
  74. 'repo_url': '/PaddlePaddle/PaddleTS.git',
  75. 'branch': 'release_v1.1',
  76. 'pkg_name': 'paddlets',
  77. 'lib_name': 'paddlets',
  78. 'pdx_pkg_name': 'PaddleTS_api',
  79. 'editable': False,
  80. 'path_env': 'PADDLE_PDX_PADDLETS_PATH',
  81. 'pdx_pkg_deps': ['pandas', 'ruamel.yaml'],
  82. },
  83. 'PaddleNLP': {
  84. 'repo_url': '/PaddlePaddle/PaddleNLP.git',
  85. 'branch': 'develop',
  86. 'pkg_name': 'paddlenlp',
  87. 'lib_name': 'paddlenlp',
  88. 'pdx_pkg_name': 'PaddleNLP_api',
  89. 'editable': False,
  90. 'path_env': 'PADDLE_PDX_PADDLENLP_PATH',
  91. },
  92. 'PaddleSpeech': {
  93. 'repo_url': '/PaddlePaddle/PaddleSpeech.git',
  94. 'branch': 'develop',
  95. 'pkg_name': 'paddlespeech',
  96. 'lib_name': 'paddlespeech',
  97. 'pdx_pkg_name': 'PaddleSpeech_api',
  98. 'editable': False,
  99. 'path_env': 'PADDLE_PDX_PADDLESPEECH_PATH',
  100. 'requires': ['PaddleNLP'],
  101. },
  102. 'PARL': {
  103. 'repo_url': '/PaddlePaddle/PARL.git',
  104. 'branch': 'develop',
  105. 'pkg_name': 'parl',
  106. 'lib_name': 'parl',
  107. 'pdx_pkg_name': 'PARL_api',
  108. 'editable': False,
  109. 'path_env': 'PADDLE_PDX_PARL_PATH',
  110. },
  111. 'PaddleMIX': {
  112. 'repo_url': '/PaddlePaddle/PaddleMIX.git',
  113. 'branch': 'develop',
  114. 'pkg_name': 'paddlemix',
  115. 'lib_name': 'paddlemix',
  116. 'pdx_pkg_name': 'PaddleMIX_api',
  117. 'editable': True,
  118. 'extra_editable': 'ppdiffusers',
  119. 'path_env': 'PADDLE_PDX_PADDLEMIX_PATH',
  120. 'requires': ['PaddleNLP'],
  121. },
  122. }
  123. def get_repo_meta(repo_name):
  124. """ get_repo_meta """
  125. return REPO_META[repo_name]
  126. def get_all_repo_names():
  127. """ get_all_repo_names """
  128. return REPO_NAMES