From 277fab991e07e51f51636a5de503c26204c1ad22 Mon Sep 17 00:00:00 2001 From: Lin Chen <87420660+lchen1019@users.noreply.github.com> Date: Sat, 2 Nov 2024 19:25:03 +0800 Subject: [PATCH] support dataset more than 255 classes --- detectron2/data/dataset_mapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/detectron2/data/dataset_mapper.py b/detectron2/data/dataset_mapper.py index a8714f7990..da00a87a20 100644 --- a/detectron2/data/dataset_mapper.py +++ b/detectron2/data/dataset_mapper.py @@ -156,7 +156,7 @@ def __call__(self, dataset_dict): # USER: Remove if you don't do semantic/panoptic segmentation. if "sem_seg_file_name" in dataset_dict: - sem_seg_gt = utils.read_image(dataset_dict.pop("sem_seg_file_name"), "L").squeeze(2) + sem_seg_gt = utils.read_image(dataset_dict.pop("sem_seg_file_name")).astype("double") else: sem_seg_gt = None