This repository has been archived by the owner on Aug 21, 2024. It is now read-only.
feat: Import yolo semantic segmentation labels #247
Closed
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.
HI!
This implements the functionality to import yolo formatted semantic segmentation labels. It generates the .xml label configuration file and the .json tasks file, just as the currently implemented object detection import.
I basically adapted the existing object detection import method on a new file to process the segmentation labels, added the corresponding subparser in the main script to extend the cli input arguments from (yolo|coco) to (yolo|yolo-seg|coco) and updated the documentation to include the new argument. I also adapted the object detection tests and added the corresponding tests for semantic segmentation.
As the label-studio contributing guide specifies, I tried to use ruff and blue for linting and coding style. Ruff raised errors on files that I didn't modify and blue also changed files that I didn't touch, so I just left them as they were and opened the PR for further guidance.