change folder structure

Author: fxia22

build.sh

@@ -4,6 +4,7 @@
 import os.path
 import torch
 import numpy as np
+import sys
 
 class PartDataset(data.Dataset):
     def __init__(self,
@@ -86,13 +87,14 @@ def __len__(self):
 
 
 if __name__ == '__main__':
+    datapath = sys.argv[1]
     print('test')
-    d = PartDataset(root = 'shapenetcore_partanno_segmentation_benchmark_v0', class_choice = ['Chair'])
+    d = PartDataset(root = datapath, class_choice = ['Chair'])
     print(len(d))
     ps, seg = d[0]
     print(ps.size(), ps.type(), seg.size(),seg.type())
 
-    d = PartDataset(root = 'shapenetcore_partanno_segmentation_benchmark_v0', classification = True)
+    d = PartDataset(root = datapath, classification = True)
     print(len(d))
     ps, cls = d[0]
     print(ps.size(), ps.type(), cls.size(),cls.type())

datasets.py renamed to pointnet/dataset.py

@@ -0,0 +1,5 @@
+SCRIPT=`realpath $0`
+SCRIPTPATH=`dirname $SCRIPT`
+echo $SCRIPTPATH
+
+g++ -std=c++11 $SCRIPTPATH/../utils/render_balls_so.cpp -o $SCRIPTPATH/../utils/render_balls_so.so -shared -fPIC -O2 -D_GLIBCXX_USE_CXX11_ABI=0

pointnet.py renamed to pointnet/model.py

@@ -1,3 +1,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 -

scripts/build.sh

@@ -0,0 +1,9 @@
+# install using 'pip install -e .'
+
+from setuptools import setup
+
+setup(name='pointnet',
+      packages=['pointnet'],
+      package_dir={'pointnet': 'pointnet'},
+      install_requires=['torch'],
+    version='0.0.1')