search

lvwj19 / PPR-Net-plus

PPR-Net++: Accurate 6D Pose Estimation in Stacked Scenarios
Apache License 2.0
36 stars 6 forks source link

Train dataset #4

Closed Junyu-Zhu closed 2 years ago

Junyu-Zhu commented 2 years ago

Hi, I've tried to generate train dataset in the file convert_to_pointcloud/bunny/generate_train_dataset.py. But it shows me the error about tensorflow like following picture libcudart I used tensorflow2.4.0, cuda11.0. Can you tell me what's the exact version of tensorflow and cuda when you generated the train dataset. And is this generated dataset the same as the dataset when you trained the pprplus net? Thanks a lot.

lvwj19 commented 2 years ago

The code can run in tensorflow1.5.0, cuda9.0. The dataset generation process for pprnet++ is the same as pprnet.

Junyu-Zhu commented 2 years ago

Ok, thanks a lot.