swarmauri / swarmauri-sdk

https://swarmauri.com
Apache License 2.0
10 stars 21 forks source link

[Feature Research]: ultralytics computer vision models #121

Open cobycloud opened 3 weeks ago

cobycloud commented 3 weeks ago

Feature Name

UltralyticsComputerVisionModel

Feature Description

Create concrete implementations to use pytorch hub to preload yolov5 models for computer vision predictions: https://docs.ultralytics.com/yolov5/tutorials/pytorch_hub_model_loading/#detailed-example

__

Research Findings

https://docs.ultralytics.com/

Potential Impact

This would add computer vision use cases to the SDK.

Additional Resources (optional)

No response

Feature Priority

Medium

Shravan773 commented 2 weeks ago

Additional Documentation for Ultralytics:

cobycloud commented 2 weeks ago

Prediction Documentation

https://docs.ultralytics.com/modes/predict/