__init__.py 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  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. from __future__ import absolute_import
  15. __version__ = '1.3.5'
  16. gui_mode = True
  17. import os
  18. if 'FLAGS_eager_delete_tensor_gb' not in os.environ:
  19. os.environ['FLAGS_eager_delete_tensor_gb'] = '0.0'
  20. if 'FLAGS_allocator_strategy' not in os.environ:
  21. os.environ['FLAGS_allocator_strategy'] = 'auto_growth'
  22. if "CUDA_VISIBLE_DEVICES" in os.environ:
  23. if os.environ["CUDA_VISIBLE_DEVICES"].count("-1") > 0:
  24. os.environ["CUDA_VISIBLE_DEVICES"] = ""
  25. import paddle
  26. version = paddle.__version__.strip().split('.')
  27. if version[0] == '1':
  28. if version[1] != '8':
  29. raise Exception(
  30. 'For running paddlex(v{}), Version of paddlepaddle should be greater than 1.8.3'.
  31. format(__version__))
  32. #import paddlehub as hub
  33. #if hub.__version__.strip().split('.')[0] > '1':
  34. # raise Exception("Try to reinstall Paddlehub by 'pip install paddlehub==1.8.2' while paddlepaddle < 2.0")
  35. if hasattr(paddle, 'enable_static'):
  36. paddle.enable_static()
  37. from .utils.utils import get_environ_info
  38. from . import cv
  39. from . import det
  40. from . import seg
  41. from . import cls
  42. from . import slim
  43. from . import converter
  44. from . import tools
  45. from . import deploy
  46. try:
  47. import pycocotools
  48. except:
  49. print(
  50. "[WARNING] pycocotools is not installed, detection model is not available now."
  51. )
  52. print(
  53. "[WARNING] pycocotools install: https://paddlex.readthedocs.io/zh_CN/develop/install.html#pycocotools"
  54. )
  55. env_info = get_environ_info()
  56. load_model = cv.models.load_model
  57. datasets = cv.datasets
  58. transforms = cv.transforms
  59. log_level = 2
  60. from . import interpret