From bdd70323a3884556af33f4baa616aba1bc66f4d8 Mon Sep 17 00:00:00 2001 From: hofee Date: Fri, 13 Sep 2024 09:40:08 +0000 Subject: [PATCH] remove unnecessary --- configs/server/split_dataset_config.yaml | 22 +++++++++++++++++++ .../strategy_generate_config.yaml | 4 ++-- configs/{ => server}/train_config.yaml | 18 +++++++-------- .../{ => server}/view_generate_config.yaml | 6 ++--- configs/split_dataset_config.yaml | 22 ------------------- core/dataset.py | 3 --- modules/gf_view_finder.py | 3 --- 7 files changed, 36 insertions(+), 42 deletions(-) create mode 100644 configs/server/split_dataset_config.yaml rename configs/{ => server}/strategy_generate_config.yaml (76%) rename configs/{ => server}/train_config.yaml (70%) rename configs/{ => server}/view_generate_config.yaml (81%) delete mode 100644 configs/split_dataset_config.yaml diff --git a/configs/server/split_dataset_config.yaml b/configs/server/split_dataset_config.yaml new file mode 100644 index 0000000..774d37d --- /dev/null +++ b/configs/server/split_dataset_config.yaml @@ -0,0 +1,22 @@ + +runner: + general: + seed: 0 + device: cpu + cuda_visible_devices: "0,1,2,3,4,5,6,7" + + experiment: + name: debug + root_dir: "experiments" + + split: + root_dir: "../data/sample_for_training/scenes" + type: "unseen_instance" # "unseen_category" + datasets: + OmniObject3d_train: + path: "../data/sample_for_training/OmniObject3d_train.txt" + ratio: 0.9 + + OmniObject3d_test: + path: "../data/sample_for_training/OmniObject3d_test.txt" + ratio: 0.1 \ No newline at end of file diff --git a/configs/strategy_generate_config.yaml b/configs/server/strategy_generate_config.yaml similarity index 76% rename from configs/strategy_generate_config.yaml rename to configs/server/strategy_generate_config.yaml index 73fae50..34a0693 100644 --- a/configs/strategy_generate_config.yaml +++ b/configs/server/strategy_generate_config.yaml @@ -25,8 +25,8 @@ runner: datasets: OmniObject3d: #"/media/hofee/data/data/temp_output" - root_dir: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/scenes" - model_dir: "/media/hofee/data/data/scaled_object_meshes" + root_dir: "../data/sample_for_training/scenes" + model_dir: "../data/scaled_object_meshes" #output_dir: "/media/hofee/data/data/label_output" diff --git a/configs/train_config.yaml b/configs/server/train_config.yaml similarity index 70% rename from configs/train_config.yaml rename to configs/server/train_config.yaml index b8d9c0b..59e0c29 100644 --- a/configs/train_config.yaml +++ b/configs/server/train_config.yaml @@ -31,18 +31,18 @@ runner: dataset: OmniObject3d_train: - root_dir: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/scenes" + root_dir: "../data/sample_for_training/scenes" source: nbv_reconstruction_dataset - split_file: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/OmniObject3d_train.txt" + split_file: "../data/sample_for_training/OmniObject3d_train.txt" ratio: 1.0 batch_size: 1 num_workers: 12 pts_num: 4096 OmniObject3d_test: - root_dir: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/scenes" + root_dir: "../data/sample_for_training/scenes" source: nbv_reconstruction_dataset - split_file: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/OmniObject3d_train.txt" + split_file: "../data/sample_for_training/OmniObject3d_train.txt" eval_list: - pose_diff ratio: 0.1 @@ -68,15 +68,15 @@ module: transformer_seq_encoder: pts_embed_dim: 1024 pose_embed_dim: 256 - num_heads: 2 # 4 - ffn_dim: 128 # 256 - num_layers: 2 # 3 - output_dim: 1024 # 2048 + num_heads: 4 + ffn_dim: 256 + num_layers: 3 + output_dim: 2048 gf_view_finder: t_feat_dim: 128 pose_feat_dim: 256 - main_feat_dim: 1024 # 2048 + main_feat_dim: 2048 regression_head: Rx_Ry_and_T pose_mode: rot_matrix per_point_feature: False diff --git a/configs/view_generate_config.yaml b/configs/server/view_generate_config.yaml similarity index 81% rename from configs/view_generate_config.yaml rename to configs/server/view_generate_config.yaml index cb5b8a8..5f928c2 100644 --- a/configs/view_generate_config.yaml +++ b/configs/server/view_generate_config.yaml @@ -7,9 +7,9 @@ runner: name: debug root_dir: experiments generate: - object_dir: /media/hofee/data/data/scaled_object_meshes - table_model_path: /media/hofee/data/data/others/table.obj - output_dir: /media/hofee/repository/nbv_reconstruction_data_512 + object_dir: ../data/scaled_object_meshes + table_model_path: ../data/others/table.obj + output_dir: ../data/nbv_reconstruction_data_512 binocular_vision: true plane_size: 10 max_views: 512 diff --git a/configs/split_dataset_config.yaml b/configs/split_dataset_config.yaml deleted file mode 100644 index 7ef58e9..0000000 --- a/configs/split_dataset_config.yaml +++ /dev/null @@ -1,22 +0,0 @@ - -runner: - general: - seed: 0 - device: cpu - cuda_visible_devices: "0,1,2,3,4,5,6,7" - - experiment: - name: debug - root_dir: "experiments" - - split: - root_dir: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/scenes" - type: "unseen_instance" # "unseen_category" - datasets: - OmniObject3d_train: - path: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/OmniObject3d_train.txt" - ratio: 0.9 - - OmniObject3d_test: - path: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/OmniObject3d_test.txt" - ratio: 0.1 \ No newline at end of file diff --git a/core/dataset.py b/core/dataset.py index b4d810a..4cfc6d9 100644 --- a/core/dataset.py +++ b/core/dataset.py @@ -3,9 +3,6 @@ from PytorchBoot.dataset import BaseDataset import PytorchBoot.stereotype as stereotype import torch -import sys -sys.path.append(r"/media/hofee/data/project/python/nbv_reconstruction/nbv_reconstruction") - from utils.data_load import DataLoadUtil from utils.pose import PoseUtil from utils.pts import PtsUtil diff --git a/modules/gf_view_finder.py b/modules/gf_view_finder.py index bc20586..fd05843 100644 --- a/modules/gf_view_finder.py +++ b/modules/gf_view_finder.py @@ -2,9 +2,6 @@ import torch import torch.nn as nn import PytorchBoot.stereotype as stereotype -import sys -sys.path.append(r"C:\Document\Local Project\nbv_rec\nbv_reconstruction") - from utils.pose import PoseUtil import modules.module_lib as mlib import modules.func_lib as flib