From 59ced4c5b36f639da36bb62991d88615711d5725 Mon Sep 17 00:00:00 2001
From: Runinho <3237686+Runinho@users.noreply.github.com>
Date: Thu, 7 Jul 2022 07:20:21 +0200
Subject: [PATCH] fix assertion in apply_category_id_mapping #28

---
 stemseg/data/generic_video_dataset_parser.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/stemseg/data/generic_video_dataset_parser.py b/stemseg/data/generic_video_dataset_parser.py
index 071382f..3fd3933 100644
--- a/stemseg/data/generic_video_dataset_parser.py
+++ b/stemseg/data/generic_video_dataset_parser.py
@@ -104,7 +104,7 @@ def filter_zero_instance_frames(self):
         self.segmentations = [self.segmentations[t] for t in t_to_keep]
 
     def apply_category_id_mapping(self, mapping):
-        assert set(mapping.keys()) == set(self.instance_categories.keys())
+        assert set(mapping.keys()) == set(self.instance_categories.values())
         self.instance_categories = {
             iid: mapping[current_cat_id] for iid, current_cat_id in self.instance_categories.items()
         }