diff --git a/docs/notebooks/serialise-detections-to-csv.ipynb b/docs/notebooks/serialise-detections-to-csv.ipynb index da42478c..4ca5806a 100644 --- a/docs/notebooks/serialise-detections-to-csv.ipynb +++ b/docs/notebooks/serialise-detections-to-csv.ipynb @@ -37,7 +37,7 @@ }, { "cell_type": "code", - "execution_count": 1, + "execution_count": 18, "metadata": { "id": "6vRfXc_Je5Ee" }, @@ -72,7 +72,7 @@ }, { "cell_type": "code", - "execution_count": 2, + "execution_count": 19, "metadata": { "id": "urRjZjh2f30v" }, @@ -120,7 +120,7 @@ }, { "cell_type": "code", - "execution_count": 3, + "execution_count": 20, "metadata": { "colab": { "base_uri": "https://localhost:8080/", @@ -161,7 +161,7 @@ }, { "cell_type": "code", - "execution_count": 5, + "execution_count": 21, "metadata": { "id": "Jpw_TzAsm2oL" }, @@ -186,7 +186,7 @@ }, { "cell_type": "code", - "execution_count": 6, + "execution_count": 22, "metadata": { "id": "-yvv6N1uTKSl" }, @@ -211,7 +211,7 @@ }, { "cell_type": "code", - "execution_count": 7, + "execution_count": 23, "metadata": { "id": "eRQiVkblapCk" }, @@ -226,7 +226,7 @@ }, { "cell_type": "code", - "execution_count": 8, + "execution_count": 24, "metadata": { "colab": { "base_uri": "https://localhost:8080/" @@ -256,7 +256,7 @@ }, { "cell_type": "code", - "execution_count": 9, + "execution_count": 25, "metadata": { "id": "bYpA9VxBsOvS" }, @@ -268,7 +268,7 @@ }, { "cell_type": "code", - "execution_count": 10, + "execution_count": 26, "metadata": { "id": "RrewOF02AYta" }, @@ -292,7 +292,7 @@ }, { "cell_type": "code", - "execution_count": 11, + "execution_count": 27, "metadata": { "colab": { "base_uri": "https://localhost:8080/", @@ -490,7 +490,7 @@ "[9029 rows x 8 columns]" ] }, - "execution_count": 11, + "execution_count": 27, "metadata": {}, "output_type": "execute_result" } @@ -511,7 +511,7 @@ }, { "cell_type": "code", - "execution_count": 13, + "execution_count": 28, "metadata": { "id": "aa24sW8FI-xV" }, @@ -566,7 +566,7 @@ }, { "cell_type": "code", - "execution_count": 15, + "execution_count": 29, "metadata": { "colab": { "base_uri": "https://localhost:8080/" @@ -12897,7 +12897,7 @@ " 79, 91, 69, 68, 35, 29, 45, 98, 97, 99, 100, 84, 101]), data={'frame_number': ['341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341', '341']})]" ] }, - "execution_count": 15, + "execution_count": 29, "metadata": {}, "output_type": "execute_result" } @@ -12929,7 +12929,7 @@ }, { "cell_type": "code", - "execution_count": 16, + "execution_count": 30, "metadata": { "id": "YRYFmKxR05xG" }, @@ -12958,7 +12958,7 @@ }, { "cell_type": "code", - "execution_count": 17, + "execution_count": 31, "metadata": { "colab": { "base_uri": "https://localhost:8080/", diff --git a/docs/notebooks/serialise-detections-to-json.ipynb b/docs/notebooks/serialise-detections-to-json.ipynb index 0f5850ce..3014a4b7 100644 --- a/docs/notebooks/serialise-detections-to-json.ipynb +++ b/docs/notebooks/serialise-detections-to-json.ipynb @@ -37,7 +37,7 @@ }, { "cell_type": "code", - "execution_count": 1, + "execution_count": 47, "metadata": { "id": "6vRfXc_Je5Ee" }, @@ -72,11 +72,20 @@ }, { "cell_type": "code", - "execution_count": null, + "execution_count": 48, "metadata": { "id": "urRjZjh2f30v" }, - "outputs": [], + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "people-walking.mp4 asset download complete. \n", + "\n" + ] + } + ], "source": [ "SOURCE_VIDEO_PATH = download_assets(VideoAssets.PEOPLE_WALKING)\n", "CONFIDENCE_THRESHOLD = 0.3\n", @@ -112,7 +121,7 @@ }, { "cell_type": "code", - "execution_count": 3, + "execution_count": 49, "metadata": { "colab": { "base_uri": "https://localhost:8080/", @@ -150,7 +159,7 @@ }, { "cell_type": "code", - "execution_count": 4, + "execution_count": 50, "metadata": { "colab": { "base_uri": "https://localhost:8080/" @@ -165,7 +174,7 @@ "VideoInfo(width=1920, height=1080, fps=25, total_frames=341)" ] }, - "execution_count": 4, + "execution_count": 50, "metadata": {}, "output_type": "execute_result" } @@ -187,7 +196,7 @@ }, { "cell_type": "code", - "execution_count": 6, + "execution_count": 51, "metadata": { "id": "Jpw_TzAsm2oL" }, @@ -212,7 +221,7 @@ }, { "cell_type": "code", - "execution_count": 7, + "execution_count": 52, "metadata": { "id": "-yvv6N1uTKSl" }, @@ -237,7 +246,7 @@ }, { "cell_type": "code", - "execution_count": 43, + "execution_count": 53, "metadata": { "id": "eRQiVkblapCk" }, @@ -252,7 +261,7 @@ }, { "cell_type": "code", - "execution_count": 44, + "execution_count": 54, "metadata": { "colab": { "base_uri": "https://localhost:8080/" @@ -273,7 +282,7 @@ }, { "cell_type": "code", - "execution_count": 45, + "execution_count": 55, "metadata": { "id": "bYpA9VxBsOvS" }, @@ -285,25 +294,11 @@ }, { "cell_type": "code", - "execution_count": 46, + "execution_count": 56, "metadata": { "id": "KXbZt-FNVnrS" }, - "outputs": [ - { - "ename": "ValueError", - "evalue": "I/O operation on closed file.", - "output_type": "error", - "traceback": [ - "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m", - "\u001b[0;31mValueError\u001b[0m Traceback (most recent call last)", - "Cell \u001b[0;32mIn[46], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[43mjson_sink\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mwrite_and_close\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n", - "File \u001b[0;32m~/rf/pr/supervision-json-csv-sink-cookbook/supervision/detection/tools/json_sink.py:90\u001b[0m, in \u001b[0;36mJSONSink.write_and_close\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m 83\u001b[0m \u001b[38;5;250m\u001b[39m\u001b[38;5;124;03m\"\"\"\u001b[39;00m\n\u001b[1;32m 84\u001b[0m \u001b[38;5;124;03mWrite and close the JSON file.\u001b[39;00m\n\u001b[1;32m 85\u001b[0m \n\u001b[1;32m 86\u001b[0m \u001b[38;5;124;03mReturns:\u001b[39;00m\n\u001b[1;32m 87\u001b[0m \u001b[38;5;124;03m None\u001b[39;00m\n\u001b[1;32m 88\u001b[0m \u001b[38;5;124;03m\"\"\"\u001b[39;00m\n\u001b[1;32m 89\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mfile:\n\u001b[0;32m---> 90\u001b[0m \u001b[43mjson\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mdump\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mdata\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mfile\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mindent\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;241;43m4\u001b[39;49m\u001b[43m)\u001b[49m\n\u001b[1;32m 91\u001b[0m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mfile\u001b[38;5;241m.\u001b[39mclose()\n", - "File \u001b[0;32m~/.pyenv/versions/3.10.14/lib/python3.10/json/__init__.py:180\u001b[0m, in \u001b[0;36mdump\u001b[0;34m(obj, fp, skipkeys, ensure_ascii, check_circular, allow_nan, cls, indent, separators, default, sort_keys, **kw)\u001b[0m\n\u001b[1;32m 177\u001b[0m \u001b[38;5;66;03m# could accelerate with writelines in some versions of Python, at\u001b[39;00m\n\u001b[1;32m 178\u001b[0m \u001b[38;5;66;03m# a debuggability cost\u001b[39;00m\n\u001b[1;32m 179\u001b[0m \u001b[38;5;28;01mfor\u001b[39;00m chunk \u001b[38;5;129;01min\u001b[39;00m iterable:\n\u001b[0;32m--> 180\u001b[0m \u001b[43mfp\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mwrite\u001b[49m\u001b[43m(\u001b[49m\u001b[43mchunk\u001b[49m\u001b[43m)\u001b[49m\n", - "\u001b[0;31mValueError\u001b[0m: I/O operation on closed file." - ] - } - ], + "outputs": [], "source": [ "json_sink.write_and_close()" ] @@ -323,7 +318,7 @@ }, { "cell_type": "code", - "execution_count": null, + "execution_count": 57, "metadata": { "colab": { "base_uri": "https://localhost:8080/", @@ -521,7 +516,7 @@ "[9029 rows x 8 columns]" ] }, - "execution_count": 12, + "execution_count": 57, "metadata": {}, "output_type": "execute_result" } @@ -542,7 +537,7 @@ }, { "cell_type": "code", - "execution_count": null, + "execution_count": 58, "metadata": { "id": "iwJf7PW6eLkU" }, @@ -597,7 +592,7 @@ }, { "cell_type": "code", - "execution_count": null, + "execution_count": 59, "metadata": { "colab": { "base_uri": "https://localhost:8080/" @@ -12928,7 +12923,7 @@ " 79, 91, 69, 68, 35, 29, 45, 98, 97, 99, 100, 84, 101]), data={'frame_number': [341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341, 341]})]" ] }, - "execution_count": 30, + "execution_count": 59, "metadata": {}, "output_type": "execute_result" } @@ -12960,7 +12955,7 @@ }, { "cell_type": "code", - "execution_count": null, + "execution_count": 60, "metadata": { "id": "I18IZQuFrPeN" }, @@ -12998,7 +12993,7 @@ }, { "cell_type": "code", - "execution_count": null, + "execution_count": 61, "metadata": { "colab": { "base_uri": "https://localhost:8080/",