File tree Expand file tree Collapse file tree 1 file changed +5
-6
lines changed
Expand file tree Collapse file tree 1 file changed +5
-6
lines changed Original file line number Diff line number Diff line change @@ -59,16 +59,15 @@ def init_detector(
5959 checkpoint_meta = checkpoint .get ('meta' , {})
6060 # save the dataset_meta in the model for convenience
6161 if 'dataset_meta' in checkpoint_meta :
62- # mmdet 3.x
63- model .dataset_meta = checkpoint_meta ['dataset_meta' ]
62+ # mmdet 3.x, all keys should be lowercase
63+ model .dataset_meta = {
64+ k .lower (): v
65+ for k , v in checkpoint_meta ['dataset_meta' ].items ()
66+ }
6467 elif 'CLASSES' in checkpoint_meta :
6568 # < mmdet 3.x
6669 classes = checkpoint_meta ['CLASSES' ]
6770 model .dataset_meta = {'classes' : classes , 'palette' : palette }
68- elif 'classes' in checkpoint_meta :
69- # < mmdet 3.x
70- classes = checkpoint_meta ['classes' ]
71- model .dataset_meta = {'classes' : classes , 'palette' : palette }
7271 else :
7372 warnings .simplefilter ('once' )
7473 warnings .warn (
You can’t perform that action at this time.
0 commit comments