Fix PALETTE AttributeError in single_gpu_test (#7230)

shinya7y · web-flow · commit 430ff0250ea0 · 2022-02-23T22:25:27.000+08:00
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():
