add sunrgbd

d622b76a · yinchimaoliang · ceb3cee7 · d622b76a
Commit d622b76a authored Jun 06, 2020 by yinchimaoliang
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

data/sunrgbd/sunrgbd_data.py data/sunrgbd/sunrgbd_data.py +4 -3

No files found.
--- a/data/sunrgbd/sunrgbd_data.py
+++ b/data/sunrgbd/sunrgbd_data.py
@@ -126,9 +126,10 @@ def extract_sunrgbd_data(idx_filename,
        print('------------- ', data_idx)
        objects = dataset.get_label_objects(data_idx)

-        if skip_empty_scene and (len(objects) == 0 or len([
-                obj for obj in objects if obj.classname in type_whitelist
-        ]) == 0):  # noqa:
+        if skip_empty_scene and \
+                (len(objects) == 0 or
+                 len([obj for obj in objects if
+                      obj.classname in type_whitelist]) == 0):
            continue

        object_list = []