From f4bf75873bdc89fb90f3810ca414861b108cc714 Mon Sep 17 00:00:00 2001 From: Daniel Saavedra Date: Thu, 16 Apr 2020 14:11:33 -0400 Subject: [PATCH] Update Example tensorflow2 --- Example_prediction.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Example_prediction.ipynb b/Example_prediction.ipynb index b02bc2e..ebe6313 100644 --- a/Example_prediction.ipynb +++ b/Example_prediction.ipynb @@ -24,7 +24,7 @@ "\n", "from utils.utils import get_yolo_boxes, makedirs\n", "from utils.bbox import draw_boxes\n", - "from keras.models import load_model\n", + "from tensorflow.keras.models import load_model\n", "from tqdm import tqdm\n", "import numpy as np\n", "import matplotlib.pyplot as plt\n", @@ -302,7 +302,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.6.8" + "version": "3.7.7" } }, "nbformat": 4,