change ds_test to ds

Author: SwanHub

notebooks/how-to-finetune-rf-detr-on-segmentation-dataset-nvidia-launchable.ipynb

@@ -624,7 +624,7 @@
       "source": [
         "import supervision as sv\n",
         "\n",
-        "ds_test = sv.DetectionDataset.from_coco(\n",
+        "ds = sv.DetectionDataset.from_coco(\n",
         "    images_directory_path=f\"{dataset.location}/test\",\n",
         "    annotations_path=f\"{dataset.location}/test/_annotations.coco.json\",\n",
         "    force_masks=True\n",
@@ -659,15 +659,15 @@
         "import matplotlib.pyplot as plt\n",
         "\n",
         "N = 16\n",
-        "L = len(ds_test)\n",
+        "L = len(ds)\n",
         "\n",
         "annotated_images = []\n",
         "\n",
         "for i in random.sample(range(L), N):\n",
-        "    path, _, annotations = ds_test[i]\n",
+        "    path, _, annotations = ds[i]\n",
         "    image = Image.open(path)\n",
         "    detections = model.predict(image, threshold=0.5)\n",
-        "    annotated_image = annotate(image, detections, classes={i: class_name for i, class_name in enumerate(ds_test.classes)})\n",
+        "    annotated_image = annotate(image, detections, classes={i: class_name for i, class_name in enumerate(ds.classes)})\n",
         "    annotated_images.append(annotated_image)\n",
         "\n",
         "fig, axes = plt.subplots(4, 4, figsize=(12, 12))\n",
@@ -734,16 +734,16 @@
         "import matplotlib.pyplot as plt\n",
         "\n",
         "N = 16\n",
-        "L = len(ds_test)\n",
+        "L = len(ds)\n",
         "\n",
         "annotated_images = []\n",
         "\n",
         "for i in random.sample(range(L), N):\n",
-        "    path, _, annotations = ds_test[i]\n",
+        "    path, _, annotations = ds[i]\n",
         "    image = Image.open(path)\n",
         "    result = model_rf.infer(image, confidence=0.5)[0]\n",
         "    detections = sv.Detections.from_inference(result)\n",
-        "    annotated_image = annotate(image, detections, classes={i: class_name for i, class_name in enumerate(ds_test.classes)})\n",
+        "    annotated_image = annotate(image, detections, classes={i: class_name for i, class_name in enumerate(ds.classes)})\n",
         "    annotated_images.append(annotated_image)\n",
         "\n",
         "fig, axes = plt.subplots(4, 4, figsize=(12, 12))\n",