Skip to content

Commit

Permalink
minor changes
Browse files Browse the repository at this point in the history
  • Loading branch information
clinplayer committed Jun 14, 2022
1 parent fe114bc commit f859b96
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 7 deletions.
3 changes: 2 additions & 1 deletion single_view_recon/.gitignore
Original file line number Diff line number Diff line change
@@ -1,2 +1,3 @@
!.gitkeep
*.pyc
*.pyc
*.txt
4 changes: 2 additions & 2 deletions single_view_recon/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ horovod==0.23.0

### Training

* Example command with required parameters to indicate the data folder:
* Example command with required parameters to indicate the data folders:
```
horovodrun -np 2 python -m src.train
--sdf_dir data/sdf-depth7-tfrecord
Expand All @@ -27,7 +27,7 @@ horovodrun -np 2 python -m src.train
* See `python -m src.train --help` for all the detailed training options.

### Testing
* Example command with required parameters to indicate the data folder and pre-trained model:
* Example command with required parameters to indicate the data folders and pre-trained model:
```
python -m src.test
--sdf_dir data/sdf-depth7-tfrecord
Expand Down
5 changes: 2 additions & 3 deletions single_view_recon/src/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
from src.utils.io_utils import save_obj_mesh_filterNAN, load_filelist
from src.utils.transform_utils import getShapenetBbox, computeOctreeSamplingPointsFromBoundingBox


def parse_args():
parser = argparse.ArgumentParser(description='3PSDF_test')
parser.add_argument('--sdf_dir', type=str, default='data/sdf-depth7-tfrecord/',
Expand All @@ -26,9 +25,9 @@ def parse_args():
parser.add_argument('--point_num', type=int, default=20000,
help='number of point samples for a single model')
parser.add_argument('--octree_depth', type=int, default=8,
help='octree depth that is consistent with the training data')
help='sampling density in the implicit field')
parser.add_argument('--load_model_path', type=str, default='weights/3psdf_svr_weights',
help='path to save the trained model')
help='path to load the pre-trained model')
parser.add_argument('--save_result_path', type=str, default='output/',
help='path to save the reconstructed meshes')
args = parser.parse_args()
Expand Down
2 changes: 1 addition & 1 deletion single_view_recon/src/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ def parse_args():
parser.add_argument('--is_finetune', type=int, default=0,
help='1 - finetune the network, 0 - train the network from scratch')
parser.add_argument('--model_weights_file', type=str, default='weights/3psdf_svr_weights',
help='path to the pretrained weights, only used if is_finetune is set to 1')
help='path to load the pretrained weights, only used if is_finetune is set to 1')
args = parser.parse_args()

return args
Expand Down

0 comments on commit f859b96

Please sign in to comment.