cleanup

weecology · Feb 17, 2024 · 5510f44 · 5510f44
1 parent 5054b96
commit 5510f44
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/docs/examples/Australia.ipynb b/docs/examples/Australia.ipynb
@@ -117,9 +117,7 @@
     }
    ],
    "source": [
-    "from deepforest import main\n",
-    "from deepforest import utilities\n",
-    "from deepforest import visualize\n",
+    "from deepforest import main, utilities, visualize\n",
     "import rasterio as rio\n",
     "from matplotlib import pyplot as plt\n",
     "import cv2\n",
@@ -129,14 +127,16 @@
     "# Make into a 3 page, remove alpha channel, and make channels last\n",
     "r = rio.open(\"/Users/benweinstein/Downloads/Plot13Ortho.tif\").read()\n",
     "r = r[:3,:,:]\n",
-    "r = r.transpose(1,2,0)"
+    "r = r.transpose(1,2,0)\n",
+    "\n",
+    "#boxes = m.predict_tile(image=r, patch_size=700, patch_overlap=0.2, iou_threshold=0.5)"
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "Great, its starts to run, but wow that's going to take too long on CPU for me to write this post. On GPU this might only take 2mins, but with a CPU almost an hour according to progress bar. Kill that and let's come back to the full prediction set when we are happy."
+    "Its starts to run, but wow that's going to take too long on CPU for me to write this post. On GPU this might only take 2mins, but with a CPU almost an hour according to progress bar. Kill that and let's come back to the full prediction set when we are happy."
    ]
   },
   {
@@ -254,7 +254,7 @@
     "gdf = utilities.boxes_to_shapefile(boxes, root_dir=\"/Users/benweinstein/Downloads\", flip_y_axis=True)\n",
     "gdf.to_file(\"/Users/benweinstein/Downloads/Plot13Ortho_crop.shp\")\n",
     "\n",
-    "plot = visualize.plot_predictions(image=r, df=gdf)\n",
+    "plot = visualize.plot_predictions(image=r, df=gdf, color=(0, 165, 255))\n",
     "plt.imshow(plot)\n",
     "plt.show()"
    ]