fix inference_demo.ipynb bug (#1236)

open-mmlab · Feb 16, 2022 · 86cc487 · 86cc487
1 parent 4a7e560
commit 86cc487
Showing 1 changed file with 32 additions and 32 deletions.
diff --git a/demo/inference_demo.ipynb b/demo/inference_demo.ipynb
@@ -3,74 +3,74 @@
   {
    "cell_type": "code",
    "execution_count": 7,
+   "source": [
+    "from mmdet3d.apis import init_model, inference_detector, show_result_meshlab"
+   ],
+   "outputs": [],
    "metadata": {
     "pycharm": {
      "is_executing": false
     }
-   },
-   "outputs": [],
-   "source": [
-    "from mmdet3d.apis import init_detector, inference_detector, show_result_meshlab"
-   ]
+   }
   },
   {
    "cell_type": "code",
    "execution_count": 8,
-   "metadata": {
-    "pycharm": {
-     "is_executing": false
-    }
-   },
-   "outputs": [],
    "source": [
     "config_file = '../configs/second/hv_second_secfpn_6x8_80e_kitti-3d-car.py'\n",
     "# download the checkpoint from model zoo and put it in `checkpoints/`\n",
     "checkpoint_file = '../work_dirs/second/epoch_40.pth'"
-   ]
+   ],
+   "outputs": [],
+   "metadata": {
+    "pycharm": {
+     "is_executing": false
+    }
+   }
   },
   {
    "cell_type": "code",
    "execution_count": 9,
+   "source": [
+    "# build the model from a config file and a checkpoint file\n",
+    "model = init_model(config_file, checkpoint_file, device='cuda:0')"
+   ],
+   "outputs": [],
    "metadata": {
     "pycharm": {
      "is_executing": false
     }
-   },
-   "outputs": [],
-   "source": [
-    "# build the model from a config file and a checkpoint file\n",
-    "model = init_detector(config_file, checkpoint_file, device='cuda:0')"
-   ]
+   }
   },
   {
    "cell_type": "code",
    "execution_count": 10,
-   "metadata": {
-    "pycharm": {
-     "is_executing": false
-    }
-   },
-   "outputs": [],
    "source": [
     "# test a single sample\n",
     "pcd = 'kitti_000008.bin'\n",
     "result, data = inference_detector(model, pcd)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 11,
+   ],
+   "outputs": [],
    "metadata": {
     "pycharm": {
      "is_executing": false
     }
-   },
-   "outputs": [],
+   }
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
    "source": [
     "# show the results\n",
     "out_dir = './'\n",
     "show_result_meshlab(data, result, out_dir)"
-   ]
+   ],
+   "outputs": [],
+   "metadata": {
+    "pycharm": {
+     "is_executing": false
+    }
+   }
   }
  ],
  "metadata": {