__init__.py 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. # copyright (c) 2020 PaddlePaddle Authors. All Rights Reserve.
  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. import os
  16. if 'FLAGS_eager_delete_tensor_gb' not in os.environ:
  17. os.environ['FLAGS_eager_delete_tensor_gb'] = '0.0'
  18. if 'FLAGS_allocator_strategy' not in os.environ:
  19. os.environ['FLAGS_allocator_strategy'] = 'auto_growth'
  20. if "CUDA_VISIBLE_DEVICES" in os.environ:
  21. if os.environ["CUDA_VISIBLE_DEVICES"].count("-1") > 0:
  22. os.environ["CUDA_VISIBLE_DEVICES"] = ""
  23. from .utils.utils import get_environ_info
  24. from . import cv
  25. from . import det
  26. from . import seg
  27. from . import cls
  28. from . import slim
  29. from . import convertor
  30. from . import tools
  31. try:
  32. import pycocotools
  33. except:
  34. print(
  35. "[WARNING] pycocotools is not installed, detection model is not available now."
  36. )
  37. print(
  38. "[WARNING] pycocotools install: https://github.com/PaddlePaddle/PaddleX/blob/develop/docs/install.md"
  39. )
  40. #import paddlehub as hub
  41. #if hub.version.hub_version < '1.6.2':
  42. # raise Exception("[ERROR] paddlehub >= 1.6.2 is required")
  43. env_info = get_environ_info()
  44. load_model = cv.models.load_model
  45. datasets = cv.datasets
  46. log_level = 2
  47. __version__ = '0.1.9.github'