Dataset for pytorch

Open CapObvios opened this issue 6 years ago • 1 comments

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.

Jul 22 '19 00:07 CapObvios

Sorry for the late response. What errors did you encounter?

Aug 10 '19 17:08 art-programmer