search

NVlabs / 6dof-graspnet

Implementation of 6-DoF GraspNet with tensorflow and python. This repo has been tested with python 2.7 and tensorflow 1.12.
MIT License
207 stars 56 forks source link

pyTorch implementation #3

Open mateoKutnjak opened 4 years ago

mateoKutnjak commented 4 years ago

Thank you for making the code public.

Is there possibility of presenting the pyTorch implementation of your code?

jsll commented 4 years ago

@mateoKutnjak as far as I know there is no PyTorch implementation of this code. I think the community would be happy if you'd make one though. Luckily you do not have to re-implement everything, but at least you need to translate model.py and train.py from Tensorflow to PyTorch.

jsll commented 4 years ago

@mateoKutnjak here is a PyTorch implementation https://github.com/jsll/pytorch_6dof-graspnet. @arsalan-mousavian I think you can close this issue now.

arsalan-mousavian commented 4 years ago

Thanks @jsll. @mateoKutnjak can close the issue.