Segmentation

[john-rocky]

This branch adds segmentation functionality to the app.

🛠️ PR Summary

_{Made with ❤️ by Ultralytics Actions}

🌟 Summary

This PR introduces segmentation models and enhances the iOS YOLO app with new features like segmentation tasks and color customization. 🚀

📊 Key Changes

• Added YOLOv8-Segment Models: Integrated segmentation models (yolov8x-seg, yolov8n-seg, etc.) alongside object detection models.
• Segmentation Task Support: Added a toggle to switch between "Detection" and "Segmentation" modes in the app interface.
• Post-Processing for Segmentation: Implemented new functionality, PostProcessSegment.swift, to handle mask generation and segmentation visualization.
• Color Enhancements: Introduced a custom Colors.swift file to align UI colors with the Ultralyics color palette.
• UI Updates: Updated storyboard for segmentation toggle and adjusted label/toolbar layout for better visuals.
• Improved CoreML Model Export: Updated the model export script to support both detection (YOLOv8) and segmentation formats.

🎯 Purpose & Impact

• Broadened Functionality: Users can now perform both object detection and segmentation, making the iOS app more powerful and versatile. 🖼️
• Enhanced Model Support: Adds seamless CoreML INT8 export compatibility for segmentation models, optimizing performance on iOS. 📱
• Improved Visualization: Segmentation masks are now displayed with customizable colors, making outputs clearer and more informative. 🎨
• User-Friendly UI: Simplified task switching via the interface and better overall layout for a smoother user experience. ✨

This update positions the YOLO iOS app as a robust tool for both object detection and instance segmentation, elevating its potential uses across various domains! 🛠️

[ambitious-octopus]

@john-rocky Can you update the README.md and specify that segmentation models should also be exported?

[john-rocky]

@ambitious-octopus
I also added conversion of the seg model to the export script in README.md.
However, for the option to download the model from GitHub Release Assets, we will need to add the seg model to your GitHub Release Assets.

[ambitious-octopus]

Now it's better, @john-rocky did you test this PR independently?

[ambitious-octopus]

Hey @john-rocky Is this ready?

[john-rocky]

@ambitious-octopus Yes, now this branch works fine.