Allow tensorflow v2 behaviour.

fizyr · Feb 10, 2020 · d569577 · d569577
1 parent a7b0ba2
commit d569577
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 8 deletions.
diff --git a/keras_maskrcnn/bin/evaluate.py b/keras_maskrcnn/bin/evaluate.py
@@ -20,7 +20,6 @@
 import os
 import sys
 
-import keras_retinanet.backend
 from keras_retinanet.utils.config import read_config_file
 from keras_retinanet.utils.gpu import setup_gpu
 from keras_retinanet.utils.keras_version import check_keras_version
@@ -36,9 +35,6 @@
 from ..preprocessing.csv_generator import CSVGenerator
 from ..utils.eval import evaluate
 
-# Disable tensorflow v2 behavior as it breaks functionality.
-keras_retinanet.backend.disable_tensorflow_v2_behavior()
-
 
 def create_generator(args):
     if args.dataset_type == 'coco':

diff --git a/keras_maskrcnn/bin/train.py b/keras_maskrcnn/bin/train.py
@@ -22,7 +22,6 @@
 
 import keras
 
-import keras_retinanet.backend
 import keras_retinanet.losses
 from keras_retinanet.callbacks import RedirectModel
 from keras_retinanet.utils.config import read_config_file, parse_anchor_parameters
@@ -42,9 +41,6 @@
 from .. import models
 from ..callbacks.eval import Evaluate
 
-# Disable tensorflow v2 behavior as it breaks functionality.
-keras_retinanet.backend.disable_tensorflow_v2_behavior()
-
 
 def model_with_weights(model, weights, skip_mismatch):
     if weights is not None: