31 lines
901 B
Python
31 lines
901 B
Python
import sys
|
|
import os
|
|
path = os.path.abspath(__file__)
|
|
for i in range(2):
|
|
path = os.path.dirname(path)
|
|
PROJECT_ROOT = path
|
|
sys.path.append(PROJECT_ROOT)
|
|
|
|
from datasets.dataset import BaseDataset
|
|
|
|
class DatasetFactory:
|
|
@staticmethod
|
|
def create(config) -> BaseDataset:
|
|
pass
|
|
|
|
|
|
''' ------------ Debug ------------ '''
|
|
if __name__ == "__main__":
|
|
|
|
from configs.config import ConfigManager
|
|
|
|
ConfigManager.load_config_with('/home/data/hofee/project/ActivePerception/ActivePerception/configs/server_train_config.yaml')
|
|
ConfigManager.print_config()
|
|
dataset = DatasetFactory.create(ConfigManager.get("settings", "test", "dataset_list")[1])
|
|
print(len(dataset))
|
|
data_test = dataset.__getitem__(107000)
|
|
print(data_test['src_path'])
|
|
import pickle
|
|
# with open("data_sample_new.pkl", "wb") as f:
|
|
# pickle.dump(data_test, f)
|