search

caizhongang / waymo_kitti_converter

A toolkit for Waymo Open Dataset <-> KITTI conversions
152 stars 30 forks source link

About small_gt_generator.py #11

Closed mrsempress closed 1 year ago

mrsempress commented 2 years ago

First of all thanks for the great tool. But I got an error while using small_gt_generator.py.

File "tools/small_gt_generator.py", line 47, in main
   frame = file_data['frame_idx']
TypeError: 'TFRecordDatasetV2' object is not subscriptable

The parameters I set are as follows:

tfrecords_load_dir = '/home/data/waymo/validation'
val_list_load_pathname = '/home/data/waymo/val.txt'
gt_load_pathname = 'gt.bin' # which is produced by gt_generator.py

gt_small_save_pathname = 'small_gt.bin'

Am I misunderstanding something?

caizhongang commented 2 years ago

Hi @mrsempress , this repo is no longer actively maintained. My best guess is that TensorFlow has changed the way TFRecordDataset behaves (as you can see the V2 in the name). I would suggest directly looking into TF's documentation to see if there are suitable methods to get a slice of the data.