Test base on pytorch-1.0 and fix bugs.

Author: delldu

README.md

@@ -8,7 +8,7 @@ This repo is implementation for PointNet(https://arxiv.org/abs/1612.00593) in py
 bash build.sh #build C++ code for visualization
 bash download.sh #download dataset
 python train_classification.py #train 3D model classification
-python python train_segmentation.py # train 3D model segmentaion
+python train_segmentation.py # train 3D model segmentaion
 
 python show_seg.py --model seg/seg_model_20.pth  # show segmentation results
 ```

show_cls.py

@@ -34,8 +34,7 @@
 
 testdataloader = torch.utils.data.DataLoader(test_dataset, batch_size=32, shuffle = True)
 
-
-classifier = PointNetCls(k = len(test_dataset.classes), num_points = opt.num_points)
+classifier = PointNetCls(k=len(test_dataset.classes))
 classifier.cuda()
 classifier.load_state_dict(torch.load(opt.model))
 classifier.eval()

train_classification.py

@@ -55,8 +55,7 @@
     pass
 
 
-classifier = PointNetCls(k = num_classes, num_points = opt.num_points)
-
+classifier = PointNetCls(k=num_classes)
 
 if opt.model != '':
     classifier.load_state_dict(torch.load(opt.model))