search

metadriverse / trafficgen

[ICRA 2023] The official code for paper "TrafficGen: Learning to Generate Diverse and Realistic Traffic Scenarios"
Apache License 2.0
153 stars 24 forks source link

Some questions about trainning #26

Closed LeonJoe13 closed 1 year ago

LeonJoe13 commented 1 year ago

It's a very meaningful and exciting work! I have several problems about cluster training

  1. dataset I download dataset from https://console.cloud.google.com/storage/browser/waymo_open_dataset_motion_v_1_1_0/uncompressed/scenario/training_20s?pageState=(%22StorageObjectListTable%22:(%22f%22:%22%255B%255D%22))&prefix=&forceOnObjectsSortingFiltering=false , which has 50 uncompressed files like image. and I stored them in patha, then I run sh utils/data_trans.sh PATH_A PATH_B, terminal shows image But I get processed data in pathb like image image I'm not sure it 's right (I don't know what none_XX.pkl mean. and every time I run sh utils/data_trans.sh again, the total number is different.
  2. Trainning After I got processed data, I modify the 'data_path' in init/configs and act/configs to PATH_B: image, run python init/uitls/init_dataset.py got error: image could you tell me how to solve these problems? Thank you!
Alan-LanFeng commented 1 year ago

  1. There are 1000 files in total. Make sure you download all of them

  2. Adding this line of code in init_dataset.py can solve it. from trafficgen.utils.typedef import AgentType, RoadEdgeType, RoadLineType

LeonJoe13 commented 1 year ago

Thank for your reply! I have fixed above problems.

there is a new problem:

When I want to connect the generated scene with Metadrive, there is a error like:

image

I have download the latest metadrive, it seems that the metadrive lack some keys for trafficgen,could you tell me how to solve? many thanks!

Alan-LanFeng commented 1 year ago

MetaDrive now only supports Scenario Description, and we have planned to update trafficgen to support that format in the future.

pengzhenghao commented 1 year ago

Yes for reproducing result of TrafficGen we suggest stick to the old version of MetaDrive Best,ZhenghaoOn Aug 19, 2023, at 04:12, Alan @.***> wrote: Closed #26 as completed.

—Reply to this email directly, view it on GitHub, or unsubscribe.You are receiving this because you are subscribed to this thread.Message ID: @.***>