Small bug fixes.

predict.py

@@ -27,8 +27,8 @@ import cv2
 from fast_pointnet_v2 import save_patches_dataset, predict_vertex_from_patch
 #from fast_voxel import predict_vertex_from_patch_voxel
 #import time
-from fast_pointnet_class_v2 import save_patches_dataset as save_patches_dataset_class
-from fast_pointnet_class_v2 import predict_class_from_patch
+from fast_pointnet_class import save_patches_dataset as save_patches_dataset_class
+from fast_pointnet_class import predict_class_from_patch
 #from fast_pointnet_class_10d import predict_class_from_patch as predict_class_from_patch_10d
 from scipy.spatial.distance import cdist
 from scipy.optimize import linear_sum_assignment