dougsm / ggcnn

Generative Grasping CNN from "Closing the Loop for Robotic Grasping: A Real-time, Generative Grasp Synthesis Approach" (RSS 2018)
BSD 3-Clause "New" or "Revised" License
485 stars 139 forks source link

memory overflow #3

Closed 1329844541 closed 5 years ago

1329844541 commented 5 years ago

Hello,I read your paper and I am more intersted in your work.But my memory overflow.I saw #2 have the same question.I wonder if you can give a tensflow version's flie ('.tfcords'),or the only way to solution is to get a bigger memory?I would be greteful to your kindness.

dougsm commented 5 years ago

Hello, as I mentioned in the other issue (https://github.com/dougsm/ggcnn/issues/2#issuecomment-455093665), you might be able to simply adjust the script so that it writes the data incrementally to the output file, rather than generating it all at once, which will get around the memory issue. Another alternative may be to generate a smaller dataset with fewer augmentations, which you can change using the parameters in the file. Hope that helps.