Update formatting.py (#10743)

open-mmlab · Aug 8, 2023 · 4d2fbb0 · 4d2fbb0
1 parent 5412e85
commit 4d2fbb0
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/mmdet/datasets/transforms/formatting.py b/mmdet/datasets/transforms/formatting.py
@@ -133,10 +133,8 @@ def transform(self, results: dict) -> dict:
 
         img_meta = {}
         for key in self.meta_keys:
-            assert key in results, f'`{key}` is not found in `results`, ' \
-                f'the valid keys are {list(results)}.'
-            img_meta[key] = results[key]
-
+            if key in results:
+                img_meta[key] = results[key]
         data_sample.set_metainfo(img_meta)
         packed_results['data_samples'] = data_sample