hofee/nbv_reconstruction
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

global_and_partial_global: upd

Browse Source
This commit is contained in:
hofee 2024-10-30 15:34:15 +00:00
parent b82b92eebb
commit 96fa40cc35
2 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
configs/server/server_split_dataset_config.yaml
View File

@ -6,17 +6,17 @@ runner:
cuda_visible_devices: "0,1,2,3,4,5,6,7"
experiment:
name: debug
name: server_split_dataset
root_dir: "experiments"
split: #
root_dir: "/data/hofee/data/packed_preprocessed_data"
root_dir: "/data/hofee/data/new_full_data"
type: "unseen_instance" # "unseen_category"
datasets:
OmniObject3d_train:
path: "/data/hofee/data/OmniObject3d_train.txt"
path: "/data/hofee/data/new_full_data_list/OmniObject3d_train.txt"
ratio: 0.9
OmniObject3d_test:
path: "/data/hofee/data/OmniObject3d_test.txt"
path: "/data/hofee/data/new_full_data_list/OmniObject3d_test.txt"
ratio: 0.1

3
core/pipeline.py
View File

@ -54,10 +54,7 @@ class NBVReconstructionPipeline(nn.Module):
return perturbed_x, random_t, target_score, std
def forward_train(self, data):
start_time = time.time()
main_feat = self.get_main_feat(data)
end_time = time.time()
print("get_main_feat time: ", end_time - start_time)
""" get std """
best_to_world_pose_9d_batch = data["best_to_world_pose_9d"]
perturbed_x, random_t, target_score, std = self.pertube_data(