Skip to content

tree-disk-segmentation v0.2.0

Choose a tag to compare

@github-actions github-actions released this 25 Feb 16:09
7cacd32
Segmentation new ai (#29)

* feat: add ultralytics and ipykernel dependencies; update .gitignore for data files

* feat: replace U2NET with YOLO for object detection; refactor segmentation pipeline

* fix: correct default class name from "logs" to "log" in polygon point extraction

* fix: jupyter now ready

* refactor: update Config class to use Path for file paths and improve type hints

* feat: update YOLO detection function to return object masks and improve logging

* fix: update model path in test for YOLO segmentation

* feat: enhance mask application to support multiple polygons and save output image

* fix: update model path in README for tree-disk-segmentation

* refactor: update run function to return masks instead of polygon points and streamline image processing

* feat: update launch configuration to use YOLO model and add debug options for segmentation

* chore: bump version to 0.2.0 in pyproject.toml

* fix: update output directory in README examples for clarity

* docs: update README to clarify segmentation output and parameters

* test: update test to assert masks instead of result for tree disk segmentation

---------

Co-authored-by: MEISSTO <tony.meissner@mercedes-benz.com>