From 5fd8ba1ae85d5f8064d89ae2a51a91068aade0a5 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 28 Apr 2019 18:25:29 +0200 Subject: [PATCH] Fixed background being saved as instance Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index 13507f8..abb2c0e 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -378,7 +378,7 @@ def prepare_scenenet_val(data_path: str, protobuf_path: str) -> Tuple[List[List[ else: continue # only save instances that are positive instances and not background - instances_traj_dict[instance_id] = instance_dict + instances_traj_dict[instance_id] = instance_dict # iterate through images/frames for view in trajectory.views: