huangfu/scenarionet
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use the utils from MetaDrive to update object summaries; update ScenarioDescription doc (#52)

Browse Source 
* Update

* update

* update

* update
This commit is contained in:
Zhenghao (Mark) Peng
2024-01-21 19:52:57 -08:00
committed by GitHub
parent 1f61a4b9d1
commit 07cb55c1cc
4 changed files with 21 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@@ -66,6 +66,11 @@ All operations and API reference is available at
our [documentation](https://scenarionet.readthedocs.io/en/latest/operations.html).
If you already have ScenarioNet installed, you can check all operations by `python -m scenarionet.list`.
## ScenarioNet dataset and Scenario Description
Please refer to the [Scenario Description section](https://metadrive-simulator.readthedocs.io/en/latest/scenario_description.html
) in MetaDrive documentation for a walk-through.
## Citation
If you used this project in your research, please cite:

4
documentation/description.rst
View File

@@ -4,4 +4,6 @@
Scenario Description
########################
Coming soon! @Zhenghao please help this.
Please visit the
`Scenario Description section <https://metadrive-simulator.readthedocs.io/en/latest/scenario_description.html>`_
in MetaDrive documentation.

14
scenarionet/converter/utils.py
View File

@@ -224,16 +224,12 @@ def write_to_directory_single_worker(
scenario_id = sd_scenario[SD.ID]
export_file_name = SD.get_export_file_name(dataset_name, dataset_version, scenario_id)
# add agents summary
summary_dict = {}
for track_id, track in sd_scenario[SD.TRACKS].items():
summary_dict[track_id] = SD.get_object_summary(state_dict=track, id=track_id)
sd_scenario[SD.METADATA][SD.SUMMARY.OBJECT_SUMMARY] = summary_dict
if hasattr(SD, "update_summaries"):
SD.update_summaries(sd_scenario)
else:
raise ValueError("Please update MetaDrive to latest version.")
# count some objects occurrence
sd_scenario[SD.METADATA][SD.SUMMARY.NUMBER_SUMMARY] = SD.get_number_summary(sd_scenario)
# update summary/mapping dicy
# update summary/mapping dict
if export_file_name in summary:
logger.warning("Scenario {} already exists and will be overwritten!".format(export_file_name))
summary[export_file_name] = copy.deepcopy(sd_scenario[SD.METADATA])

16
tutorial/read_established_scenarionet_dataset.ipynb
View File

@@ -232,8 +232,8 @@
{
"data": {
"text/plain": [
"\u001b[0;31mSignature:\u001b[0m \u001b[0mread_dataset_summary\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mdataset_path\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
"\u001b[0;31mDocstring:\u001b[0m\n",
"\u001B[0;31mSignature:\u001B[0m \u001B[0mread_dataset_summary\u001B[0m\u001B[0;34m(\u001B[0m\u001B[0mdataset_path\u001B[0m\u001B[0;34m)\u001B[0m\u001B[0;34m\u001B[0m\u001B[0;34m\u001B[0m\u001B[0m\n",
"\u001B[0;31mDocstring:\u001B[0m\n",
"Read the dataset and return the metadata of each scenario in this dataset.\n",
"\n",
"Args:\n",
@@ -244,8 +244,8 @@
" 1) the summary dict mapping from scenario ID to its metadata,\n",
" 2) the list of all scenarios IDs, and\n",
" 3) a dict mapping from scenario IDs to the folder that hosts their files.\n",
"\u001b[0;31mFile:\u001b[0m ~/scenarionet/scenarionet/common_utils.py\n",
"\u001b[0;31mType:\u001b[0m function"
"\u001B[0;31mFile:\u001B[0m ~/scenarionet/scenarionet/common_utils.py\n",
"\u001B[0;31mType:\u001B[0m function"
]
},
"metadata": {},
@@ -3593,8 +3593,8 @@
{
"data": {
"text/plain": [
"\u001b[0;31mSignature:\u001b[0m \u001b[0mread_scenario\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mdataset_path\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mmapping\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mscenario_file_name\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
"\u001b[0;31mDocstring:\u001b[0m\n",
"\u001B[0;31mSignature:\u001B[0m \u001B[0mread_scenario\u001B[0m\u001B[0;34m(\u001B[0m\u001B[0mdataset_path\u001B[0m\u001B[0;34m,\u001B[0m \u001B[0mmapping\u001B[0m\u001B[0;34m,\u001B[0m \u001B[0mscenario_file_name\u001B[0m\u001B[0;34m)\u001B[0m\u001B[0;34m\u001B[0m\u001B[0;34m\u001B[0m\u001B[0m\n",
"\u001B[0;31mDocstring:\u001B[0m\n",
"Read a scenario pkl file and return the Scenario Description instance.\n",
"\n",
"Args:\n",
@@ -3604,8 +3604,8 @@
"\n",
"Returns:\n",
" The Scenario Description instance of that scenario.\n",
"\u001b[0;31mFile:\u001b[0m ~/scenarionet/scenarionet/common_utils.py\n",
"\u001b[0;31mType:\u001b[0m function"
"\u001B[0;31mFile:\u001B[0m ~/scenarionet/scenarionet/common_utils.py\n",
"\u001B[0;31mType:\u001B[0m function"
]
},
"metadata": {},