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" cuda_visible_devices: "0,1,2,3,4,5,6,7"
experiment: experiment:
name: debug name: server_split_dataset
root_dir: "experiments" root_dir: "experiments"
split: # split: #
root_dir: "/data/hofee/data/packed_preprocessed_data" root_dir: "/data/hofee/data/new_full_data"
type: "unseen_instance" # "unseen_category" type: "unseen_instance" # "unseen_category"
datasets: datasets:
OmniObject3d_train: OmniObject3d_train:
path: "/data/hofee/data/OmniObject3d_train.txt" path: "/data/hofee/data/new_full_data_list/OmniObject3d_train.txt"
ratio: 0.9 ratio: 0.9
OmniObject3d_test: OmniObject3d_test:
path: "/data/hofee/data/OmniObject3d_test.txt" path: "/data/hofee/data/new_full_data_list/OmniObject3d_test.txt"
ratio: 0.1 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 return perturbed_x, random_t, target_score, std
def forward_train(self, data): def forward_train(self, data):
start_time = time.time()
main_feat = self.get_main_feat(data) main_feat = self.get_main_feat(data)
end_time = time.time()
print("get_main_feat time: ", end_time - start_time)
""" get std """ """ get std """
best_to_world_pose_9d_batch = data["best_to_world_pose_9d"] best_to_world_pose_9d_batch = data["best_to_world_pose_9d"]
perturbed_x, random_t, target_score, std = self.pertube_data( perturbed_x, random_t, target_score, std = self.pertube_data(