SDK: Improve the PyTorch adapter layer #5455
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
SpecLad
force-pushed
the
pytorch-improvements
branch
from
014d039
to
0438dc1
Compare
This was referenced Dec 13, 2022
Merged
…ve types This is more consistent with torchvision models, and lets the `ExtractBoundingBoxes` outputs to be used directly with the `torchmetrics` package (or at least the `MeanAveragePrecision` class).
…mapping The default numbering is generally unpredictable, so it's not especially useful outside of basic use cases. Given that there's no way to specify custom indexes for labels on the server side, make it possible on the client side. The input dictionary maps label _names_ to indexes, because names is what a CVAT user would be familiar with. The output dictionary in the `Target` class still maps label _IDs_, because that's what CVAT uses in the annotation data models.
SpecLad
force-pushed
the
pytorch-improvements
branch
from
0438dc1
to
d06e041
Compare
mikhail-treskin
pushed a commit
to retailnext/cvat
that referenced
this pull request
Jul 1, 2023
* Make the extractors return tensors instead of Python data structures. * Let the user specify custom label IDs.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
See the commit messages for more details.
Motivation and context
This adds functionality that I found to be missing while making a demo script for an object detection scenario.
How has this been tested?
Unit tests and the aforementioned demo script.
Checklist
developbranch[ ] I have added a description of my changes into CHANGELOG file[ ] I have linked related issues (read github docs)[ ] I have increased versions of npm packages if it is necessary (cvat-canvas,cvat-core, cvat-data and cvat-ui)
License
Feel free to contact the maintainers if that's a concern.