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

adjust display table position

Browse Source
This commit is contained in:
hofee 2024-09-19 11:44:10 +00:00
parent bb75372f7e
commit 55684e86ba
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
core/dataset.py
View File

@ -7,7 +7,7 @@ from PytorchBoot.utils.log_util import Log
import torch import torch
import os import os
import sys import sys
sys.path.append(r"/media/hofee/data/project/python/nbv_reconstruction/nbv_reconstruction") sys.path.append(r"/home/data/hofee/project/nbv_rec/nbv_reconstruction")
from utils.data_load import DataLoadUtil from utils.data_load import DataLoadUtil
from utils.pose import PoseUtil from utils.pose import PoseUtil
@ -17,7 +17,6 @@ from utils.reconstruction import ReconstructionUtil
@stereotype.dataset("nbv_reconstruction_dataset") @stereotype.dataset("nbv_reconstruction_dataset")
class NBVReconstructionDataset(BaseDataset): class NBVReconstructionDataset(BaseDataset):
DISPLAY_TABLE_POSITION = np.asarray([0,0,0.85])
def __init__(self, config): def __init__(self, config):
super(NBVReconstructionDataset, self).__init__(config) super(NBVReconstructionDataset, self).__init__(config)
self.config = config self.config = config
@ -206,15 +205,16 @@ if __name__ == "__main__":
torch.manual_seed(seed) torch.manual_seed(seed)
np.random.seed(seed) np.random.seed(seed)
config = { config = {
"root_dir": "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/scenes", "root_dir": "../data/sample_for_training/scenes",
"split_file": "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/OmniObject3d_train.txt", "model_dir": "../data/scaled_object_meshes",
"model_dir": "/media/hofee/data/data/scaled_object_meshes", "source": "nbv_reconstruction_dataset",
"split_file": "../data/sample_for_training/OmniObject3d_train.txt",
"ratio": 0.5, "ratio": 0.5,
"batch_size": 2, "batch_size": 2,
"filter_degree": 75, "filter_degree": 75,
"num_workers": 0, "num_workers": 0,
"pts_num": 32684, "pts_num": 4096,
"type": namespace.Mode.TEST, "type": namespace.Mode.TRAIN,
} }
ds = NBVReconstructionDataset(config) ds = NBVReconstructionDataset(config)
print(len(ds)) print(len(ds))

2
utils/data_load.py
View File

@ -6,7 +6,7 @@ import trimesh
from utils.pts import PtsUtil from utils.pts import PtsUtil
class DataLoadUtil: class DataLoadUtil:
DISPLAY_TABLE_POSITION = np.asarray([0,0,0.85]) DISPLAY_TABLE_POSITION = np.asarray([0,0,0.895])
@staticmethod @staticmethod
def get_path(root, scene_name, frame_idx): def get_path(root, scene_name, frame_idx):
path = os.path.join(root, scene_name, f"{frame_idx}") path = os.path.join(root, scene_name, f"{frame_idx}")