push to origin

Author: skim2257

.gitmodules

@@ -0,0 +1,29 @@
+[submodule "sampling_and_meshing/PoissonRecon"]
+	path = sampling_and_meshing/PoissonRecon
+	url = https://github.com/mkazhdan/PoissonRecon
+	branch = master
+[submodule "sampling_and_meshing/O-CNN"]
+	path = sampling_and_meshing/O-CNN
+	url = https://github.com/palanglois/O-CNN.git
+	branch = master
+[submodule "extension"]
+	path = extension
+	url = https://github.com/ThibaultGROUEIX/chamfer_pytorch
+[submodule "ChamferDistancePytorch"]
+	path = ChamferDistancePytorch
+	url = https://github.com/ThibaultGROUEIX/ChamferDistancePytorch
+[submodule "vcglib"]
+	path = vcglib
+	url = https://github.com/cnr-isti-vclab/vcglib.git
+[submodule "auxiliary/ChamferDistancePytorch"]
+	path = auxiliary/ChamferDistancePytorch
+	url = https://github.com/ThibaultGROUEIX/ChamferDistancePytorch
+[submodule "auxiliary/sampling_and_meshing/O-CNN"]
+	path = auxiliary/sampling_and_meshing/O-CNN
+	url = https://github.com/palanglois/O-CNN.git
+[submodule "auxiliary/sampling_and_meshing/PoissonRecon"]
+	path = auxiliary/sampling_and_meshing/PoissonRecon
+	url = https://github.com/mkazhdan/PoissonRecon.git
+[submodule "auxiliary/netvision"]
+	path = auxiliary/netvision
+	url = https://github.com/palanglois/netvision.git

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2019 ThibaultGROUEIX
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

chamfer.sh

@@ -0,0 +1 @@
+/cluster/home/sejinkim/miniconda3/envs/atlasnet/bin/nvcc -I/cluster/home/sejinkim/miniconda3/envs/atlasnet/lib/python3.6/site-packages/torch/include -I/cluster/home/sejinkim/miniconda3/envs/atlasnet/lib/python3.6/site-packages/torch/include/torch/csrc/api/include -I/cluster/home/sejinkim/miniconda3/envs/atlasnet/lib/python3.6/site-packages/torch/include/TH -I/cluster/home/sejinkim/miniconda3/envs/atlasnet/lib/python3.6/site-packages/torch/include/THC -I/cluster/home/sejinkim/miniconda3/envs/atlasnet/include/python3.6m -c -c /cluster/home/sejinkim/projects/AtlasNet/auxiliary/ChamferDistancePytorch/chamfer3D/chamfer3D.cu -o /cluster/home/sejinkim/projects/AtlasNet/build/temp.linux-x86_64-3.6/auxiliary/ChamferDistancePytorch/chamfer3D/chamfer3D.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=chamfer_3D -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_61,code=compute_61 -gencode=arch=compute_61,code=sm_61 -std=c++14

models.md

@@ -0,0 +1,11 @@
+2214155: 02021-04-03T16:25:08.091652
+* 9 hexagons
+
+2214150: 02021-04-03T16:16:07.162028
+* 9 hexagons
+
+2214157: 02021-04-03T16:25:39.079993
+* 9 squares
+
+2214148: 02021-04-03T16:12:40.192672
+* 9 squares

requirements.txt

@@ -0,0 +1,12 @@
+easydict
+joblib
+matplotlib
+ninja
+numpy
+scipy
+termcolor
+tornado
+urllib3
+visdom
+websocket-client
+https://github.com/PyMesh/PyMesh/releases/download/v0.2.1/pymesh2-0.2.1-cp36-cp36m-linux_x86_64.whl

test.sh

@@ -0,0 +1 @@
+python train.py --demo --demo_input_path dataset/data/ShapeNetV1PointCloud/02691156/10155655850468db78d106ce0a280f87.points.ply.npy --reload_model_path log/02021-04-03T16\:25\:08.091652/network.pth

train.py

@@ -0,0 +1,46 @@
+import sys
+import auxiliary.argument_parser as argument_parser
+import auxiliary.my_utils as my_utils
+import time
+import torch
+from auxiliary.my_utils import yellow_print
+
+"""
+Main training script.
+author : Thibault Groueix 01.11.2019
+"""
+
+opt = argument_parser.parser()
+torch.cuda.set_device(opt.multi_gpu[0])
+#torch.cuda.set_device(0)
+my_utils.plant_seeds(random_seed=opt.random_seed)
+import training.trainer as trainer
+
+trainer = trainer.Trainer(opt)
+trainer.build_dataset()
+trainer.build_network()
+trainer.build_optimizer()
+trainer.build_losses()
+trainer.start_train_time = time.time()
+
+if opt.demo:
+    with torch.no_grad():
+        trainer.demo(opt.demo_input_path)
+    sys.exit(0)
+
+if opt.run_single_eval:
+    with torch.no_grad():
+        trainer.test_epoch()
+    sys.exit(0)
+
+for epoch in range(trainer.epoch, opt.nepoch):
+    trainer.train_epoch()
+    with torch.no_grad():
+        trainer.test_epoch()
+    trainer.dump_stats()
+    trainer.increment_epoch()
+    trainer.save_network()
+
+yellow_print(f"Visdom url http://localhost:{trainer.opt.visdom_port}/")
+yellow_print(f"Netvision report url http://localhost:{trainer.opt.http_port}/{trainer.opt.dir_name}/index.html")
+yellow_print(f"Training time {(time.time() - trainer.start_time)//60} minutes.")