small updates

utils/generate_figures.py

@@ -5,12 +5,11 @@ sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/..')
 import numpy as np
 import cmapy as cm
 
-from utils import remove_long_edges
 
 from custom_utils.custom_utils import save_stl, write_obj
 from data.cloth3d_io import readPC2Frame
 from data.utils import Frame
-from custom_utils.custom_utils import _maskFaces
+from custom_utils.custom_utils import _maskFaces, remove_long_edges
 import plotly.graph_objects as go
 import plotly.express as ex
 from plotly.subplots import make_subplots

utils/postprocess.py

@@ -14,7 +14,7 @@ import plotly.graph_objects as go
 import plotly.express as ex
 from pytorch3d.structures import Meshes
 
-from custom_utils.custom_utils import _maskFaces
+from custom_utils.custom_utils import _maskFaces, remove_long_edges
 from losses.losses import _nearest_neighbour