transformers
fc5401d1 - Return logits from semantic segmentation post-process (#46163)

Commit
42 days ago
Return logits from semantic segmentation post-process (#46163) * Add return_logits * Add SemanticSegmentationPostProcessOutput * update copyright date * rename to PostProcessorOutput * update ConditionalDetr * align docstrings * update Detr * update Dpt * update Eomt * update Mask2former * update MobileNet * update MobileVit * update Oneformer * update Sam3 * update VideoEomt * update SegGpt * unify variable names * speed up semantic_map computation * remove SemanticSegmentationPostProcessorOutputs from global __init__ * add tests * remove extra tests * fix Eomt tests * inherit from BatchFeature * add PostProcessSemanticSegmentationTestMixin * fix test * remove dataclass from PostProcessorOutput * add post processor docs * add comment * Update src/transformers/image_processing_outputs.py Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com> * Update docs/source/en/add_vision_processing_components.md Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com> * Update docs/source/en/add_vision_processing_components.md Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com> * Update docs/source/en/add_vision_processing_components.md Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com> * Update docs/source/en/add_vision_processing_components.md Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com> * Update docs/source/en/add_vision_processing_components.md Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com> --------- Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com> Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Author
Parents
Loading