pytorch_6dof-graspnet icon indicating copy to clipboard operation
pytorch_6dof-graspnet copied to clipboard

Published 20 hours ago • verified publisher icon jsll

Weird VAE results

Open bearpaw opened this issue 3 years ago • 4 comments

Hi @jsll , I found the VAE results are pretty weird compared with the GAN results. Here are the examples:

python -m demo.main --generate_dense_grasps --num_grasp_samples 20 --grasp_sampler_folder checkpoints/vae_pretrained/

image image

python -m demo.main --grasp_sampler_folder checkpoints/vae_pretrained/

image image image Any ideas?

bearpaw avatar May 04 '21 00:05 bearpaw

Interesting findings. As far as I know, this also happened to me sometimes with the VAE network but not this severely. I should have uploaded the correct VAE so the problem should not be there.

What I do know, is that I did not train the PyTorch VAE for nearly as long of a time as the TensorFlow one, I think I just trained it for a couple of days in contract to a week or so for the TensorFlow model. I would bet most of the problem stems from that fact.

jsll avatar May 04 '21 06:05 jsll

Thanks Jens! That makes sense.

Just to clarify that the results I showed were generated from the pretrained model you provided. Would you be able to upload the correct VAE once you get a chance? Thanks!

bearpaw avatar May 04 '21 16:05 bearpaw

I actually think that should be the pretrained model, at least I did not find anyone performing better. If I have time end of this week, I might setup a new training round and leave it training for a longer time.

jsll avatar May 04 '21 18:05 jsll

Thanks! I was wondering there might be some potential bugs that generated results like these.

bearpaw avatar May 04 '21 18:05 bearpaw