Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
46 changes: 46 additions & 0 deletions src/av2/evaluation/detection/SUBMISSION_FORMAT.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,46 @@
# 3D Object Detection Submission Format

The evaluation expects the following fields within a `pandas.DataFrame`:

- `tx_m`: x-component of the object translation in the egovehicle reference frame.
- `ty_m`: y-component of the object translation in the egovehicle reference frame.
- `tz_m`: z-component of the object translation in the egovehicle reference frame.
- `length_m`: Object extent along the x-axis in meters.
- `width_m`: Object extent along the y-axis in meters.
- `height_m`: Object extent along the z-axis in meters.
- `qw`: Real quaternion coefficient.
- `qx`: First quaternion coefficient.
- `qy`: Second quaternion coefficient.
- `qz`: Third quaternion coefficient.
- `score`: Object confidence.
- `log_id`: Log id associated with the detection.
- `timestamp_ns`: Timestamp associated with the detection.
- `category`: Object category.

An example looks like this:

```python
# These detections are only for example purposes.

display(detections) # Detections is type `pd.DataFrame`
tx_m ty_m tz_m length_m width_m height_m qw qx qy qz score log_id timestamp_ns category
0 -162.932968 1.720428 0.039064 1.596262 0.772320 1.153996 0.125843 0.0 0.0 0.992050 0.127634 b0116f1c-f88f-3c09-b4bf-fc3c8ebeda56 315968193659921000 WHEELCHAIR
1 -120.362213 19.875946 -0.382618 1.441901 0.593825 1.199819 0.802836 0.0 0.0 0.596200 0.126565 b0116f1c-f88f-3c09-b4bf-fc3c8ebeda56 315968193659921000 BICYCLE
...
14000000 10.182907 29.489899 0.662969 9.166531 1.761454 1.615999 0.023469 0.0 0.0 -0.999725 0.322177 b2d9d8a5-847b-3c3b-aed1-c414319d20af 315978610360111000 REGULAR_VEHICLE

detections.columns
Index(['tx_m', 'ty_m', 'tz_m', 'length_m', 'width_m', 'height_m', 'qw', 'qx',
'qy', 'qz', 'score', 'log_id', 'timestamp_ns', 'category'],
dtype='object')
```

We need to export the above dataframe for submission. This can be done by:

```python
import pandas as pd

detections.to_feather("detections.feather")
```

Lastly, submit this file to the competition leaderboard.
32 changes: 30 additions & 2 deletions tutorials/object_detection_evaluation.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,24 @@
"source": [
"## Preparing detections for submission.\n",
"\n",
"Our submission server expects a `.feather` file that contains a list of detections for the entire test set."
"The evaluation expects the following 14 fields within a `pandas.DataFrame`:\n",
"\n",
"- `tx_m`: x-component of the object translation in the egovehicle reference frame.\n",
"- `ty_m`: y-component of the object translation in the egovehicle reference frame.\n",
"- `tz_m`: z-component of the object translation in the egovehicle reference frame.\n",
"- `length_m`: Object extent along the x-axis in meters.\n",
"- `width_m`: Object extent along the y-axis in meters.\n",
"- `height_m`: Object extent along the z-axis in meters.\n",
"- `qw`: Real quaternion coefficient.\n",
"- `qx`: First quaternion coefficient.\n",
"- `qy`: Second quaternion coefficient.\n",
"- `qz`: Third quaternion coefficient.\n",
"- `score`: Object confidence.\n",
"- `log_id`: Log id associated with the detection.\n",
"- `timestamp_ns`: Timestamp associated with the detection.\n",
"- `category`: Object category.\n",
"\n",
"Additional details can be found in [SUBMISSION_FORMAT.md](../src/av2/evaluation/detection/SUBMISSION_FORMAT.md)."
]
},
{
Expand All @@ -85,6 +102,17 @@
"source": [
"display(metrics)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Finally, if you would like to submit to the evaluation server, you just need to export your detections into a `.feather` file. This can be done by:\n",
"\n",
"```python\n",
"dts.to_feather(\"detections.feather\")\n",
"```"
]
}
],
"metadata": {
Expand All @@ -106,7 +134,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.2"
"version": "3.9.12"
},
"orig_nbformat": 4
},
Expand Down