Speed up semantic segmentation annotations loading #205
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.
This PR includes:
SegmentationAnnotation.combine_to_numpy()
function.Loader benchmarks (1 worker, 5000 images):
unassigned_pixels = np.sum(array, axis=0) == 0
- 3:15 (previous)unassigned_pixels = ~np.any(array, axis=0)
- 1:58 (new)SegmentationAnnotation.combine_to_numpy()
- 1:45Training benchmarks (1 epoch, 5000 images, 8 workers):