NielsRogge
|
979b039c89
|
Add DPT (#15991)
* First draft
* More improvements
* Add fusion blocks
* Make conversion script work for dpt_large
* Make conversion script work
* Improve implementation
* Improve conversion script
* Add DPTForSemanticSegmentation
* Make conversion work for semantic segmentation
* Add tests
* Remove print statements
* First draft
* Redesign neck
* Improve tests
* Improve implementation some more
* Make neck output list of tensors
* Improve neck and feature extractor
* Fix integration tests
* Make more tests pass
* Make all tests pass
* Add missing config archive map
* Add in_index attribute to make heads accept list of tensors
* Apply suggestions from code review
* Apply suggestions from code review
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
* Apply some more suggestions
* Add copied from statements
* Remove assert
* Apply suggestions from code review
* Apply suggestions from code review
* Remove DPTInterpolate in favor of nn.Upsample
* Add comments
* Apply suggestions from code review
* Apply suggestions from code review
* Add proposed design
* Update design
* Add DPTReassembleLayer
* Add DPTFeatureFusionStage
* Apply more suggestions from code review
* Apply suggestions from code review
* Apply suggestions from code review
* Fix rebase
* Update in_index and out_indices
* Fix conversion script
* Fix code quality
* Add model to toctree and use DepthEstimatorOutput
* Fix rebase
* Fix code examples
* Improve code
* Fix copied from statements
* Apply suggestions from code review
* Remove compute_loss method
* Apply suggestions from code review
* Fix documentation tests file
* Remove test.py file
* Improve doc example
Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Co-authored-by: Niels Rogge <nielsrogge@nielss-mbp.home>
|
2022-03-28 16:28:10 +02:00 |
|