Hello!
Thank you for your impressive work!

When I was trying to train the pytorch model, I converted the ".tfrecords" dataset using the script you provided. However, the pytorch/train_planenet.py seems to be expecting a little bit different dataset.
Could you please provide the code for using the dataset converted from ".tfrecords" if it exists?
Otherwise, could you please tell how to obtain the dataset the pytorch/datasets scripts expect?

I would be extremely grateful if you could help me.