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

global_only: config

Browse Source
This commit is contained in:
hofee 2024-10-29 12:18:51 +00:00
parent f533104e4a
commit 2487039445
2 changed files with 1 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
configs/server/server_train_config.yaml
View File

@ -78,7 +78,7 @@ dataset:
pipeline: pipeline:
nbv_reconstruction_local_pts_pipeline: nbv_reconstruction_pipeline:
modules: modules:
pts_encoder: pointnet_encoder pts_encoder: pointnet_encoder
seq_encoder: transformer_seq_encoder seq_encoder: transformer_seq_encoder
@ -87,25 +87,6 @@ pipeline:
eps: 1e-5 eps: 1e-5
global_scanned_feat: True global_scanned_feat: True
nbv_reconstruction_global_pts_pipeline:
modules:
pts_encoder: pointnet_encoder
pose_seq_encoder: transformer_seq_encoder
pose_encoder: pose_encoder
view_finder: gf_view_finder
eps: 1e-5
global_scanned_feat: True
nbv_reconstruction_global_pts_n_num_pipeline:
modules:
pts_encoder: pointnet_encoder
transformer_seq_encoder: transformer_seq_encoder
pose_encoder: pose_encoder
view_finder: gf_view_finder
pts_num_encoder: pts_num_encoder
eps: 1e-5
global_scanned_feat: True
module: module:

1
core/pipeline.py
View File

@ -29,7 +29,6 @@ class NBVReconstructionPipeline(nn.Module):
self.eps = float(self.config["eps"]) self.eps = float(self.config["eps"])
self.enable_global_scanned_feat = self.config["global_scanned_feat"]
def forward(self, data): def forward(self, data):
mode = data["mode"] mode = data["mode"]