21 Commits

Author SHA1 Message Date
Zhenghao Peng
6cda061ed8 Fix a bug in Waymo conversion: GPU should be disable (#64)
* Update waymo.rst

* Update waymo.rst

* allow generate all data

* update readme

* update

* better logging info

* more info

* up

* fix

* add note on GPU

* better log

* format
2024-02-20 13:28:09 -08:00
Zhenghao (Mark) Peng
07cb55c1cc Use the utils from MetaDrive to update object summaries; update ScenarioDescription doc (#52)
* Update

* update

* update

* update
2024-01-21 19:52:57 -08:00
Quanyi Li
2db22c74a8 Update nuScenes & Waymo Optimization (#47)
* update can bus

* Create LICENSE

* update waymo doc

* protobuf requirement

* just warning

* Add warning for proto

* update PR template

* fix length bug

* try sharing nusc

* imu heading

* fix 161 168

* add badge

* fix doc

* update doc

* format

* update cp

* update nuscenes interface

* update doc

* prediction nuscenes

* use drivable aread for nuscenes

* allow converting prediction

* format

* fix bug

* optimize

* clean RAM

* delete more

* restore to

* add only lane

* use token

* add warning

* format

* fix bug
2023-12-02 13:50:44 +00:00
Quanyi Li
3dd161188c Update the citation in README.md (#40) 2023-10-24 19:22:27 +01:00
PENG Zhenghao
63a4c38e7f Update Citation information (#37) 2023-10-23 07:10:04 -07:00
PENG Zhenghao
af9fe0a2aa Update document, add a colab example for reading data, upgrade numpy dependency (#34)
* Minor update to docs

* WIP

* adjust numpy requirement

* prepare example for reading data from SN dataset

* prepare example for reading data from SN dataset

* clean
2023-10-18 14:32:01 -07:00
Quanyi Li
403af36ba1 Update readme (#24)
* update figure

* add colab to doc
2023-09-12 22:28:21 +01:00
Quanyi Li
931b549555 Update README.md (#23) 2023-09-12 22:13:03 +01:00
Quanyi Li
1e410b2b98 Colab exp (#22)
* add example

* add new workflow

* fix bug

* pull asset automatically

* add colab

* fix test

* add colab to readme
2023-09-12 22:10:42 +01:00
Quanyi Li
4affa68a06 Update doc (#20)
* Add list.py and desc

* add operations

* add structure

* update readme

* format

* update readme
2023-08-26 22:29:54 +01:00
Quanyi Li
e9d816ddd8 Doc-example (#18)
* reactive traffic example

* structure

* structure

* waymo example

* rename and add doc

* finish example

* example

* start from 2
2023-08-25 18:35:46 +01:00
QuanyiLi
86e0113921 doc 2023-08-08 18:18:33 +01:00
Quanyi Li
e9c1419a91 update readme (#9) 2023-07-03 13:50:33 +01:00
Quanyi Li
5f5a5b9531 Update env (#7)
* add capture script

* gymnasium API

* training with gymnasium API
2023-06-23 19:50:40 +01:00
Quanyi Li
88b4faa00f Update video/webpage/code 2023-06-22 12:17:20 +01:00
Quanyi Li
9acdcaf321 Make video (#5)
* generate accident scene

* construction PG

* no object

* accident prob

* capture script

* update nuscenes toolds

* make video

* format

* fix test

* update readme

* update readme

* format

* format
2023-06-17 17:19:40 +01:00
Quanyi Li
db50bca7fd Add come updates for Neurips paper (#4)
* scenarionet training

* wandb

* train utils

* fix callback

* run PPO

* use pg test

* save path

* use torch

* add dependency

* update ignore

* update training

* large model

* use curriculum training

* add time to exp name

* storage_path

* restore

* update training

* use my key

* add log message

* check seed

* restore callback

* restore call bacl

* add log message

* add logging message

* restore ray1.4

* length 500

* ray 100

* wandb

* use tf

* more levels

* add callback

* 10 worker

* show level

* no env horizon

* callback result level

* more call back

* add diffuculty

* add mroen stat

* mroe stat

* show levels

* add callback

* new

* ep len 600

* fix setup

* fix stepup

* fix to 3.8

* update setup

* parallel worker!

* new exp

* add callback

* lateral dist

* pg dataset

* evaluate

* modify config

* align config

* train single RL

* update training script

* 100w eval

* less eval to reveal

* 2000 env eval

* new trianing

* eval 1000

* update eval

* more workers

* more worker

* 20 worker

* dataset to database

* split tool!

* split dataset

* try fix

* train 003

* fix mapping

* fix test

* add waymo tqdm

* utils

* fix bug

* fix bug

* waymo

* int type

* 8 worker read

* disable

* read file

* add log message

* check existence

* dist 0

* int

* check num

* suprass warning

* add filter API

* filter

* store map false

* new

* ablation

* filter

* fix

* update filyter

* reanme to from

* random select

* add overlapping checj

* fix

* new training sceheme

* new reward

* add waymo train script

* waymo different config

* copy raw data

* fix bug

* add tqdm

* update readme

* waymo

* pg

* max lateral dist 3

* pg

* crash_done instead of penalty

* no crash done

* gpu

* update eval script

* steering range penalty

* evaluate

* finish pg

* update setup

* fix bug

* test

* fix

* add on line

* train nuplan

* generate sensor

* udpate training

* static obj

* multi worker eval

* filx bug

* use ray for testing

* eval!

* filter senario

* id filter

* fox bug

* dist = 2

* filter

* eval

* eval ret

* ok

* update training pg

* test before use

* store data=False

* collect figures

* capture pic

---------

Co-authored-by: Quanyi Li <quanyi@bolei-gpu02.cs.ucla.edu>
2023-06-10 18:56:33 +01:00
QuanyiLi
1c01e4417e rename scripts and update readme 2023-05-19 14:03:48 +01:00
Quanyi Li
8d0fc77aa5 Update README.md 2023-05-19 13:56:36 +01:00
Quanyi Li
00ad6f7e61 Readme (#3)
* rename to merge dataset

* add -d for operation

* test move

* add move function

* test remove

* format

* dataset -> database

* add readme

* format.sh

* test assert

* rename to database in .sh
2023-05-19 13:51:18 +01:00
Quanyi Li
69a7cfa841 Initial commit 2023-05-05 19:20:08 +01:00