convert .off files to .ply

.gitignore

@@ -8,3 +8,4 @@ shapenetcore_partanno_segmentation_benchmark_v0/
 cls/
 seg/
 *.egg-info/
+ModelNet40/

pointnet/dataset.py

@@ -1,8 +1,10 @@
-from __future__ import print_function
+# from __future__ import print_function
 import torch.utils.data as data
 import os
 import os.path
 import torch
+import meshio
+from pathlib import Path
 import numpy as np
 import sys
 from tqdm import tqdm 
@@ -145,7 +147,8 @@ def __init__(self,
                  root,
                  npoints=2500,
                  split='train',
-                 data_augmentation=True):
+                 data_augmentation=True,
+                 convert_off_to_ply=True):
         self.npoints = npoints
         self.root = root
         self.split = split
@@ -164,6 +167,29 @@ def __init__(self,
         print(self.cat)
         self.classes = list(self.cat.keys())
 
+        if convert_off_to_ply:
+            p = Path(self.root)
+            print('searching path', p, 'to convert .off files to .ply')
+            l = list(p.glob('**/*.off'))
+            for in_file in l:
+                # print(in_file)
+                with in_file.open() as f:
+                    # some OFF files in original dataset had OFF345 345 344 where 
+                    # OFF collided with the number. Needs \n
+                    lines = f.readlines()
+                if lines[0] != 'OFF\n':
+                    # print(lines[0:3])
+                    lines[0] = lines[0][0:3] + '\n' + lines[0][3:]
+                    # print(lines[0:3])
+                    with in_file.open('w') as f:
+                        lines = "".join(lines)
+                        f.write(lines)
+
+                mesh = meshio.read(in_file, file_format="off")
+                out_file = in_file.replace(in_file.with_suffix('.ply'))
+                # print(out_file)
+                mesh.write(out_file, file_format='ply')
+
     def __getitem__(self, index):
         fn = self.fns[index]
         cls = self.cat[fn.split('/')[0]]

scripts/download.sh

@@ -2,7 +2,8 @@ SCRIPT=`realpath $0`
 SCRIPTPATH=`dirname $SCRIPT`
 
 cd $SCRIPTPATH/..
-wget https://shapenet.cs.stanford.edu/ericyi/shapenetcore_partanno_segmentation_benchmark_v0.zip --no-check-certificate
-unzip shapenetcore_partanno_segmentation_benchmark_v0.zip
-rm shapenetcore_partanno_segmentation_benchmark_v0.zip
-cd -
+wget http://modelnet.cs.princeton.edu/ModelNet40.zip --no-check-certificate
+unzip ModelNet40.zip
+# rm ModelNet40.zip # in case you need to reuse
+cd ..
+cp test.txt trainval.txt ModelNet40/