Fix acces to a wrong key in checkpoint (open-mmlab#734)

Author: sovrasov

mmdet/apis/inference.py

@@ -34,7 +34,7 @@ def init_detector(config, checkpoint=None, device='cuda:0'):
     if checkpoint is not None:
         checkpoint = load_checkpoint(model, checkpoint)
         if 'CLASSES' in checkpoint['meta']:
-            model.CLASSES = checkpoint['meta']['classes']
+            model.CLASSES = checkpoint['meta']['CLASSES']
         else:
             warnings.warn('Class names are not saved in the checkpoint\'s '
                           'meta data, use COCO classes by default.')

tools/train.py

@@ -79,7 +79,7 @@ def main():
         cfg.checkpoint_config.meta = dict(
             mmdet_version=__version__,
             config=cfg.text,
-            classes=train_dataset.CLASSES)
+            CLASSES=train_dataset.CLASSES)
     # add an attribute for visualization convenience
     model.CLASSES = train_dataset.CLASSES
     train_detector(