Skip to content

GitLab

Commit d622b76a authored by yinchimaoliang's avatar yinchimaoliang
Browse files

add sunrgbd

parent ceb3cee7
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
data/sunrgbd/sunrgbd_data.py
View file @ d622b76a
...@@ -126,9 +126,10 @@ def extract_sunrgbd_data(idx_filename, ...@@ -126,9 +126,10 @@ def extract_sunrgbd_data(idx_filename,
print('------------- ', data_idx) print('------------- ', data_idx)
objects = dataset.get_label_objects(data_idx) objects = dataset.get_label_objects(data_idx)
if skip_empty_scene and (len(objects) == 0 or len([ if skip_empty_scene and \
obj for obj in objects if obj.classname in type_whitelist (len(objects) == 0 or
]) == 0): # noqa: len([obj for obj in objects if
obj.classname in type_whitelist]) == 0):
continue continue
object_list = [] object_list = []
......
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