"...git@developer.sourcefind.cn:OpenDAS/mmpretrain.git" did not exist on "b6df0d33815d47760d250e033ea5962deb89ae53"
Unverified Commit 2d00f82c authored by Martin Hahner's avatar Martin Hahner Committed by GitHub
Browse files

Update database_sampler.py (#715)

Fix https://github.com/open-mmlab/OpenPCDet/issues/702
parent 4f0e1f02
...@@ -19,6 +19,8 @@ class DataBaseSampler(object): ...@@ -19,6 +19,8 @@ class DataBaseSampler(object):
self.db_infos = {} self.db_infos = {}
for class_name in class_names: for class_name in class_names:
self.db_infos[class_name] = [] self.db_infos[class_name] = []
self.use_shared_memory = sampler_cfg.get('USE_SHARED_MEMORY', False)
for db_info_path in sampler_cfg.DB_INFO_PATH: for db_info_path in sampler_cfg.DB_INFO_PATH:
db_info_path = self.root_path.resolve() / db_info_path db_info_path = self.root_path.resolve() / db_info_path
...@@ -28,8 +30,7 @@ class DataBaseSampler(object): ...@@ -28,8 +30,7 @@ class DataBaseSampler(object):
for func_name, val in sampler_cfg.PREPARE.items(): for func_name, val in sampler_cfg.PREPARE.items():
self.db_infos = getattr(self, func_name)(self.db_infos, val) self.db_infos = getattr(self, func_name)(self.db_infos, val)
self.use_shared_memory = sampler_cfg.get('USE_SHARED_MEMORY', False)
self.gt_database_data_key = self.load_db_to_shared_memory() if self.use_shared_memory else None self.gt_database_data_key = self.load_db_to_shared_memory() if self.use_shared_memory else None
self.sample_groups = {} self.sample_groups = {}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment