Training GQN with different dataset

ogroth / tf-gqn

Tensorflow implementation of Neural Scene Representation and Rendering

Apache License 2.0

188 stars 35 forks source link

Training GQN with different dataset #25

Closed waiyc closed 5 years ago

waiyc commented 5 years ago

Hi,

I am trying to train the GQN with shepard_metzler_5_parts dataset. After the training has finished, I used the script mentioned in here to visualize the result. However, the generated image is just a black image.

So, the default training parameters is only for rooms_ring_camera? Do I need to change some of the training parameters in gqn_params.py for different dataset?

ogroth commented 5 years ago

Hi @waiyc , please check out the latest update on the master branch. GQN now trains fine on all datasets and I have added jupyter notebooks for easier deployment of trained networks in prediction mode.