pedestrian trajectory files required for model(.pkl)

ykotseruba / JAAD

Annotation data for JAAD (Joint Attention in Autonomous Driving) Dataset

http://data.nvision2.eecs.yorku.ca/JAAD_dataset

MIT License

166 stars 55 forks source link

pedestrian trajectory files required for model(.pkl) #21

Closed adibisoyi closed 1 month ago

adibisoyi commented 6 months ago

Hi @ykotseruba , For my model I need to extract pedestrian trajectory files as .pkl files (eg. 00131_0_109_608b.pkl). Can you help on how to generate the pickle files from the images extracted from the videos.

ykotseruba commented 6 months ago

Hi @adibisoyi, please see the _get_trajectories function (lines 1040-1117 in jaad_data.py'). This function outputs a dictionary with lists of images, coordinates, ids, etc. corresponding to the individual trajectories. For example,bbox_seq[i]' will contain bounding boxes and image_seq[i]' will contain images for the pedestrian with id stored inpids_seq[i]'.