"git@developer.sourcefind.cn:yangql/googletest.git" did not exist on "675686a139a731a2c796633e67e9421792363709"
Commit 5904f8e3 authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

remove warning from yaml and dbsampler (#121)

* remove warning from yaml and dbsampler
parent 4337fc4e
...@@ -51,9 +51,9 @@ def merge_new_config(config, new_config): ...@@ -51,9 +51,9 @@ def merge_new_config(config, new_config):
if '_BASE_CONFIG_' in new_config: if '_BASE_CONFIG_' in new_config:
with open(new_config['_BASE_CONFIG_'], 'r') as f: with open(new_config['_BASE_CONFIG_'], 'r') as f:
try: try:
yaml_config = yaml.load(f)
except:
yaml_config = yaml.load(f, Loader=yaml.FullLoader) yaml_config = yaml.load(f, Loader=yaml.FullLoader)
except:
yaml_config = yaml.load(f)
config.update(EasyDict(yaml_config)) config.update(EasyDict(yaml_config))
for key, val in new_config.items(): for key, val in new_config.items():
...@@ -70,9 +70,9 @@ def merge_new_config(config, new_config): ...@@ -70,9 +70,9 @@ def merge_new_config(config, new_config):
def cfg_from_yaml_file(cfg_file, config): def cfg_from_yaml_file(cfg_file, config):
with open(cfg_file, 'r') as f: with open(cfg_file, 'r') as f:
try: try:
new_config = yaml.load(f)
except:
new_config = yaml.load(f, Loader=yaml.FullLoader) new_config = yaml.load(f, Loader=yaml.FullLoader)
except:
new_config = yaml.load(f)
merge_new_config(config=config, new_config=new_config) merge_new_config(config=config, new_config=new_config)
......
...@@ -164,7 +164,7 @@ class DataBaseSampler(object): ...@@ -164,7 +164,7 @@ class DataBaseSampler(object):
""" """
gt_boxes = data_dict['gt_boxes'] gt_boxes = data_dict['gt_boxes']
gt_names = data_dict['gt_names'] gt_names = data_dict['gt_names'].astype(str)
existed_boxes = gt_boxes existed_boxes = gt_boxes
total_valid_sampled_dict = [] total_valid_sampled_dict = []
for class_name, sample_group in self.sample_groups.items(): for class_name, sample_group in self.sample_groups.items():
......
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