Removed some more unnecessary stuff

janblumenkamp · Sep 8, 2019 · c0f3232 · c0f3232
1 parent 27c2eb4
commit c0f3232
Showing 1 changed file with 1 addition and 24 deletions.
diff --git a/train_keras.ipynb b/train_keras.ipynb
@@ -394,7 +394,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "for i in range(0, 5):\n",
+    "for i in range(0, 3):\n",
     "    img = np.array(dataset.get_image(i))\n",
     "    \n",
     "    fig,ax = plt.subplots(1)\n",
@@ -878,29 +878,6 @@
     "    #print(\"res\", quat_res, trans_res)\n"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "## Perform a orientation determination on the validation data set"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "for imgs, masks in validation_generator:\n",
-    "    for img, mask in zip(imgs, masks[0]):        \n",
-    "        try:\n",
-    "            quat_res, trans_res = predict_pose(model, img, mask, True)\n",
-    "        except cv2.error as e:\n",
-    "            continue\n",
-    "        print(\"res\", quat_res, trans_res)\n",
-    "        print(\"=\"*30)\n"
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {},