07244707a74ad207a1b61497011b454cc3ade192,keras_rcnn/backend/__init__.py,,,#,3

Before Change

from .common import *

if keras.backend.backend() == "theano":
    from .theano_backend import *
elif keras.backend.backend() == "tensorflow":
    from .tensorflow_backend import *

After Change

_BACKEND = "tensorflow"

if "KERAS_BACKEND" in os.environ:
    _backend = os.environ["KERAS_BACKEND"]

    backends = {
        "cntk",
        "tensorflow",
        "theano"
    }

    assert _backend in backends

    _BACKEND = _backend

if _BACKEND == "cntk":
    from .cntk_backend import *
elif _BACKEND == "theano":
    from .theano_backend import *

Instances