[Fix]: fix dataset meta (#9536)

open-mmlab · Dec 27, 2022 · 99d501f · 99d501f
1 parent a9f46b6
commit 99d501f
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/mmdet/apis/inference.py b/mmdet/apis/inference.py
@@ -59,16 +59,15 @@ def init_detector(
         checkpoint_meta = checkpoint.get('meta', {})
         # save the dataset_meta in the model for convenience
         if 'dataset_meta' in checkpoint_meta:
-            # mmdet 3.x
-            model.dataset_meta = checkpoint_meta['dataset_meta']
+            # mmdet 3.x, all keys should be lowercase
+            model.dataset_meta = {
+                k.lower(): v
+                for k, v in checkpoint_meta['dataset_meta'].items()
+            }
         elif 'CLASSES' in checkpoint_meta:
             # < mmdet 3.x
             classes = checkpoint_meta['CLASSES']
             model.dataset_meta = {'classes': classes, 'palette': palette}
-        elif 'classes' in checkpoint_meta:
-            # < mmdet 3.x
-            classes = checkpoint_meta['classes']
-            model.dataset_meta = {'classes': classes, 'palette': palette}
         else:
             warnings.simplefilter('once')
             warnings.warn(