SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Restructure DETR post-processing, return prediction scores (#19262)

Browse Source 
* Restructure DetrFeatureExtractor post-processing methods
* Update post_process_instance_segmentation and post_process_panoptic_segmentation methods to return prediction scores
* Update DETR models docs
This commit is contained in:
Alara Dirik
2022-10-03 12:02:51 +03:00
committed by GitHub
parent 5cd16f01db
commit 36f52e9593
3 changed files with 161 additions and 153 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docs/source/en/model_doc/detr.mdx
View File

@@ -171,9 +171,9 @@ mean Average Precision (mAP) and Panoptic Quality (PQ). The latter objects are i
[[autodoc]] DetrFeatureExtractor
- __call__
- pad_and_create_pixel_mask
- post_process
- post_process_segmentation
- post_process_panoptic
- post_process_semantic_segmentation
- post_process_instance_segmentation
- post_process_panoptic_segmentation
## DetrModel