Skip to content

fix: sem_seg_gt.squeeze(2) crashes on 2D detectron2 grayscale images - #65

Open
andrewwhitecdw wants to merge 1 commit into
NVlabs:mainfrom
andrewwhitecdw:bugfix/dataset-mapper-da98199b
Open

fix: sem_seg_gt.squeeze(2) crashes on 2D detectron2 grayscale images#65
andrewwhitecdw wants to merge 1 commit into
NVlabs:mainfrom
andrewwhitecdw:bugfix/dataset-mapper-da98199b

Conversation

@andrewwhitecdw

Copy link
Copy Markdown

Problem

fix: sem_seg_gt.squeeze(2) crashes on 2D detectron2 grayscale images

Fix

Replace:

            if self.seg_format == "L":
                sem_seg_gt = sem_seg_gt.squeeze(2)

with:

            if self.seg_format == "L" and sem_seg_gt.ndim == 3:
                sem_seg_gt = sem_seg_gt.squeeze(2)

Files changed

  • odise/data/dataset_mapper.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant