Fixed import names

diegovalsesia · Dec 20, 2022 · ef56a73 · ef56a73
1 parent 4bceb57
commit ef56a73
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 12 deletions.
diff --git a/classification_ScanObject/models/dgcnn_hmodel.py b/classification_ScanObject/models/dgcnn_hmodel.py
@@ -16,9 +16,9 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from modelsnet.ops.manifold_layers import GeodesicLayer, MobiusLayer, LogZero, ExpZero, GyroplaneConvLayer
-from modelsnet.manifolds import PoincareBall, Euclidean
-import modelsnet.ops.nn as hypnn
+from models.ops.manifold_layers import GeodesicLayer, MobiusLayer, LogZero, ExpZero, GyroplaneConvLayer
+from models.manifolds import PoincareBall, Euclidean
+import models.ops.nn as hypnn
 
 
 def knn(x, k):

diff --git a/classification_ScanObject/models/hpointmlp.py b/classification_ScanObject/models/hpointmlp.py
@@ -4,9 +4,9 @@
 import torch.nn.functional as F
 # from torch import einsum
 # from einops import rearrange, repeat
-from modelsnet.ops.manifold_layers import GeodesicLayer, MobiusLayer, LogZero, ExpZero, GyroplaneConvLayer
-from modelsnet.manifolds import PoincareBall, Euclidean
-import modelsnet.ops.nn as hypnn
+from models.ops.manifold_layers import GeodesicLayer, MobiusLayer, LogZero, ExpZero, GyroplaneConvLayer
+from models.manifolds import PoincareBall, Euclidean
+import models.ops.nn as hypnn
 import random
 from pointnet2_ops import pointnet2_utils
 

diff --git a/classification_ScanObject/models/pointmlp.py b/classification_ScanObject/models/pointmlp.py
@@ -4,9 +4,9 @@
 import torch.nn.functional as F
 # from torch import einsum
 # from einops import rearrange, repeat
-from modelsnet.ops.manifold_layers import GeodesicLayer, MobiusLayer, LogZero, ExpZero, GyroplaneConvLayer
-from modelsnet.manifolds import PoincareBall, Euclidean
-import modelsnet.ops.nn as hypnn
+from models.ops.manifold_layers import GeodesicLayer, MobiusLayer, LogZero, ExpZero, GyroplaneConvLayer
+from models.manifolds import PoincareBall, Euclidean
+import models.ops.nn as hypnn
 
 from pointnet2_ops import pointnet2_utils
 

diff --git a/classification_ScanObject/models/pointmlp_modnet40.py b/classification_ScanObject/models/pointmlp_modnet40.py
@@ -4,9 +4,9 @@
 import torch.nn.functional as F
 # from torch import einsum
 # from einops import rearrange, repeat
-from modelsnet.ops.manifold_layers import GeodesicLayer, MobiusLayer, LogZero, ExpZero, GyroplaneConvLayer
-from modelsnet.manifolds import PoincareBall, Euclidean
-import modelsnet.ops.nn as hypnn
+from models.ops.manifold_layers import GeodesicLayer, MobiusLayer, LogZero, ExpZero, GyroplaneConvLayer
+from models.manifolds import PoincareBall, Euclidean
+import models.ops.nn as hypnn
 
 from pointnet2_ops import pointnet2_utils