Update Example tensorflow2

2020-04-16 14:11:33 -04:00
parent 4351f33609
commit f4bf75873b
1 changed files with 2 additions and 2 deletions
--- 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,