qinglew / PCN-PyTorch

Implementation of PCN(Point Completion Network) in PyTorch.
141 stars 33 forks source link

Pretrained Model #4

Closed ccc870206 closed 3 years ago

ccc870206 commented 3 years ago

Hi, it's very great work on the implementation of PCN in PyTorch! I want to further study this work. Therefore, is it possible that you also provide your pretrained model? Many thanks.

liuhuaijjin commented 3 years ago

Expect the author to publish a pretraining model

qinglew commented 3 years ago

I published a pretrained model in directory log/, please check it! In fact, according to the visualization of completion result, the pretrained model is not good enough, and I only trianed with the category of car. When I trained the model, the batch size is very small because I have only a low level GPU 2060. Maybe you can train the model with bigger batch size.

liuhuaijjin commented 3 years ago

I am very glad that you can reply to me, the recent GRNet network effect may be a little better than PCN