Fix PALETTE AttributeError in single_gpu_test (#7230)

open-mmlab · Feb 23, 2022 · 430ff02 · 430ff02
1 parent 37a1150
commit 430ff02
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdet/apis/test.py b/mmdet/apis/test.py
@@ -22,7 +22,7 @@ def single_gpu_test(model,
     model.eval()
     results = []
     dataset = data_loader.dataset
-    PALETTE = dataset.PALETTE
+    PALETTE = getattr(dataset, 'PALETTE', None)
     prog_bar = mmcv.ProgressBar(len(dataset))
     for i, data in enumerate(data_loader):
         with torch.no_grad():