There is no generate_target_course function in plan_utils.py

georgeliu233 / OPGP

[ITSC'23] Code for 'Occupancy Prediction-Guided Neural Planner for Autonomous Driving'

60 stars 3 forks source link

There is no generate_target_course function in plan_utils.py #4

Closed soyaCat closed 4 months ago

soyaCat commented 1 year ago

Hi I want to run your propocess.py python file but. I can't do it In your 'preprocess.py', it tries to import all functions from 'utils.train_utils.py'. However, 'utils.train_utils.py' attempts to call 'generate_target_course' function which is not defined in 'utils.plan_utils.py'. As a result, it seems that 'preprocess.py' does not work. Can you make a correction on this?

ZikangXiong commented 11 months ago

Same issue here. generate_target_course is missing.

WeiXiCZ commented 11 months ago

+1😢

georgeliu233 commented 11 months ago

Extremely sorry for the inconvenience!

I added the generate_target_course function.

If you have any questions, don't hesitate to reach me!

ZikangXiong commented 11 months ago

Thanks for the reply and fixing the issue!

A quick note on running

python preprocess.py \
--root_dir path/to/your/Waymo_Dataset/scenario/ \
--save_dir path/to/your/processed_data/ \
--processes=16

Under the path/to/your/processed_data/, we should create two sub-folders named train/ and valid/ at the beginning. Otherwise, it will throw out errors. Maybe considering adding os.makedirs