search

art-programmer / PlaneNet

PlaneNet: Piece-wise Planar Reconstruction from a Single RGB Image
http://art-programmer.github.io/planenet.html
MIT License
400 stars 84 forks source link

Dataset for pytorch #25

Open CapObvios opened 5 years ago

CapObvios commented 5 years ago

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.

art-programmer commented 5 years ago

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