Re-enables loading of PointNet models

Re-enables loading of PointNet and PointNet class models.
Updates model paths to the correct location.
This allows the training process to use the pre-trained models as intended.

train.py

@@ -32,10 +32,12 @@ show_visu = False
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 #pnet_model = load_pointnet_model(model_path="/home/skvrnjan/personal/hoho_pnet/initial_epoch_100.pth", device=device, predict_score=True)
-pnet_model = None
+pnet_model = load_pointnet_model(model_path="/mnt/personal/skvrnjan/hoho_pnet/initial_epoch_100.pth", device=device, predict_score=True)
+#pnet_model = None
 
 #pnet_class_model = load_pointnet_class_model(model_path="/home/skvrnjan/personal/hoho_pnet_edges_v2/initial_epoch_100.pth", device=device)
-pnet_class_model = None
+pnet_class_model = load_pointnet_class_model(model_path="/mnt/personal/skvrnjan/hoho_pnet_edges_v2/initial_epoch_100.pth", device=device)
+#pnet_class_model = None
 
 #voxel_model = load_3dcnn_model(model_path="/home/skvrnjan/personal/hoho_voxel/initial_epoch_100.pth", device=device, predict_score=True)
 voxel_model = None