You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

load_vision_backend.py 463 B

4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
12345678910111213141516
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. from __future__ import absolute_import, division, print_function
  4. from tensorlayer.backend.ops.load_backend import BACKEND
  5. if BACKEND == 'tensorflow':
  6. from .tensorflow_vision import *
  7. elif BACKEND == 'mindspore':
  8. from .mindspore_vision import *
  9. elif BACKEND == 'dragon':
  10. pass
  11. elif BACKEND == 'paddle':
  12. from .paddle_vision import *
  13. else:
  14. raise NotImplementedError("This backend is not supported")

TensorLayer3.0 是一款兼容多种深度学习框架为计算后端的深度学习库。计划兼容TensorFlow, Pytorch, MindSpore, Paddle.