transformers
27c90174 - Fix post_process_semantic_segmentation removing valid class in Conditional DETR (#42681)

Commit
109 days ago
Fix post_process_semantic_segmentation removing valid class in Conditional DETR (#42681) * Fix post_process_semantic_segmentation removing valid class in Conditional DETR Fixes #42679 Conditional DETR does not have a null class unlike DETR, so the `[..., :-1]` slice was incorrectly removing the last valid class. The fix removes this slice in both the regular and fast image processors since Conditional DETR's class_labels_classifier outputs num_labels classes (not num_labels + 1 like DETR). * Trigger CI --------- Co-authored-by: yonigozlan <yoni.gozlan@huggingface.co>
Parents
Loading