From a68572926c111e51ad3f2eeabd9508c6c72cf8a5 Mon Sep 17 00:00:00 2001
From: Vladislav Sovrasov <sovrasov.vlad@gmail.com>
Date: Wed, 29 May 2019 14:19:58 +0300
Subject: [PATCH] Fix acces to a wrong key in checkpoint (#734)
---
mmdet/apis/inference.py | 2 +-
tools/train.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/mmdet/apis/inference.py b/mmdet/apis/inference.py
index 222fc83..c8c37ed 100644
--- a/mmdet/apis/inference.py
+++ b/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.')
diff --git a/tools/train.py b/tools/train.py
index 3e06d6f..b8f21d1 100644
--- a/tools/train.py
+++ b/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(
--
GitLab