training_extensions
training_extensions copied to clipboard
Enable instance segmentation data pipeline
Summary
This PR includes:
- [x] Update/Add transforms to enable masks and polygons for iseg
- [x] Resize
- [x] Pad
- [x] RandomFlip
- [x] CachedMosaic
- [x] CachedMixUp
- [x] RandomResize
- [x] RandomCrop
- [x] FilterAnnotations
- [x] Update recipes
- [x] maskrcnn_efficientnetb2b
- [x] maskrcnn_r50
- [x] maskrcnn_swint
- [x] rtmdet_inst_tiny
- [x] Update base dataset to avoid
Noneresults
How to test
Checklist
- [ ] I have added unit tests to cover my changes.
- [ ] I have added integration tests to cover my changes.
- [ ] I have ran e2e tests and there is no issues.
- [ ] I have added the description of my changes into CHANGELOG in my target branch (e.g., CHANGELOG in develop).
- [ ] I have updated the documentation in my target branch accordingly (e.g., documentation in develop).
- [ ] I have linked related issues.
License
- [ ] I submit my code changes under the same Apache License that covers the project. Feel free to contact the maintainers if that's a concern.
- [ ] I have updated the license header for each file (see an example below).
# Copyright (C) 2024 Intel Corporation
# SPDX-License-Identifier: Apache-2.0