tests/maskformer/test_modeling_maskformer.py · 040c11f6dac72bc3088498aa19184da677563424 · chenpangpang / transformers

Tests for MaskFormerFeatureExtractor's post_process*** methods (#15929) · 040c11f6

Francesco Saverio Zuppichini authored Mar 04, 2022



* proper tests for post_process*** methods in feature extractor

* mask th == 0

* Update tests/maskformer/test_feature_extraction_maskformer.py
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* make style
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

040c11f6

test_modeling_maskformer.py 16.1 KB

Replace test_modeling_maskformer.py