From e8160bbc5df2a07638e9767fce9666cf2b42e544 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 11 Jun 2026 01:38:52 +0200 Subject: [PATCH 1/3] Improve README accuracy and links --- README.md | 64 +++++++++++++++++++++++++++++----------------- yolo11n/README.md | 2 +- yoloe11s/README.md | 2 +- 3 files changed, 43 insertions(+), 25 deletions(-) diff --git a/README.md b/README.md index 2069daa..1bb4786 100644 --- a/README.md +++ b/README.md @@ -2,9 +2,9 @@ # ๐Ÿš€ Ultralytics Replicate -Deploy YOLO11n to Replicate at https://replicate.com/ultralytics/yolo11n with ready-to-use Cog configuration and automated CI/CD workflow. +Deploy Ultralytics YOLO models to Replicate with ready-to-use Cog configurations and automated CI/CD workflows. -[![Push YOLO11n to Replicate](https://github.com/ultralytics/replicate/actions/workflows/push.yml/badge.svg)](https://github.com/ultralytics/replicate/actions/workflows/push.yml) +[![Push YOLO to Replicate](https://github.com/ultralytics/replicate/actions/workflows/push.yml/badge.svg)](https://github.com/ultralytics/replicate/actions/workflows/push.yml) [![Ultralytics Actions](https://github.com/ultralytics/replicate/actions/workflows/format.yml/badge.svg)](https://github.com/ultralytics/replicate/actions/workflows/format.yml) [![Codecov](https://codecov.io/github/ultralytics/replicate/branch/main/graph/badge.svg)](https://app.codecov.io/github/ultralytics/replicate) @@ -16,7 +16,8 @@ Deploy YOLO11n to Replicate at https://replicate.com/ultralytics/yolo11n with re ## ๐Ÿ—‚๏ธ Repository Structure -This repository provides optimized Replicate deployment for the YOLO11n model with automated CI/CD workflow. +This repository provides optimized Replicate deployments for YOLO11n, YOLOv8s WorldV2, and YOLOE-11S models with an +automated CI/CD workflow. ```plaintext ultralytics/replicate/ @@ -25,13 +26,22 @@ ultralytics/replicate/ โ”‚ โ”œโ”€โ”€ cog.yaml # Cog configuration โ”‚ โ”œโ”€โ”€ predict.py # Prediction interface โ”‚ โ””โ”€โ”€ README.md # Model documentation +โ”œโ”€โ”€ yolov8s-worldv2/ # YOLOv8s WorldV2 model deployment +โ”‚ โ”œโ”€โ”€ cog.yaml # Cog configuration +โ”‚ โ”œโ”€โ”€ predict.py # Prediction interface +โ”‚ โ””โ”€โ”€ README.md # Model documentation +โ”œโ”€โ”€ yoloe11s/ # YOLOE-11S model deployment +โ”‚ โ”œโ”€โ”€ cog.yaml # Cog configuration +โ”‚ โ”œโ”€โ”€ predict.py # Prediction interface +โ”‚ โ””โ”€โ”€ README.md # Model documentation +โ”œโ”€โ”€ assets/ # Sample images for workflow smoke tests โ”‚ โ”œโ”€โ”€ .github/workflows/ # Automated deployment โ”‚ โ”œโ”€โ”€ push.yml # Model deployment workflow โ”‚ โ”œโ”€โ”€ ci.yml # Code quality checks โ”‚ โ””โ”€โ”€ format.yml # Code formatting โ”‚ -โ”œโ”€โ”€ test_prediction.py # Local testing utility +โ”œโ”€โ”€ test_prediction.py # Local YOLO11n testing utility โ”œโ”€โ”€ requirements.txt # Dependencies โ”œโ”€โ”€ LICENSE # AGPL-3.0 license โ””โ”€โ”€ README.md # This file @@ -39,19 +49,26 @@ ultralytics/replicate/ ## โšก Quick Start -### Deploy YOLO11n Model +### Deploy a Model -Model will deploy to https://replicate.com/ultralytics/yolo11n: +Models deploy to the corresponding Replicate endpoints: ```bash # Clone repository git clone https://github.com/ultralytics/replicate.git cd replicate -# Deploy to Replicate +# Deploy YOLO11n cd yolo11n cog login cog push r8.im/ultralytics/yolo11n + +# Or deploy another configured model +cd ../yolov8s-worldv2 +cog push r8.im/ultralytics/yolov8s-worldv2 + +cd ../yoloe11s +cog push r8.im/ultralytics/yoloe-11s ``` ### Automated Deployment with GitHub Actions @@ -61,8 +78,8 @@ cog push r8.im/ultralytics/yolo11n - Add `REPLICATE_API_TOKEN` with your [Replicate API token](https://replicate.com/auth/token) 2. **Deploy:** - - **Manual**: Actions tab โ†’ "Push YOLO11n to Replicate" โ†’ Run workflow - - **Automatic**: Push changes to `main` branch auto-deploys + - **Manual**: Actions tab โ†’ "Push YOLO to Replicate" โ†’ Run workflow + - **Automatic**: Push changes to `main` builds, tests, and deploys each configured model ## ๐Ÿ› ๏ธ Installation @@ -79,22 +96,22 @@ For local development and testing: pip install -r requirements.txt ``` -## ๐ŸŽฏ YOLO11n Model +## ๐ŸŽฏ Available Models -- **Purpose**: Official YOLO11n object detection -- **Parameters**: 2.6M parameters -- **Classes**: 80 COCO classes -- **Performance**: 39.5 mAP50-95 on COCO dataset -- **Speed**: Optimized for real-time inference +| Directory | Replicate model | Predictor | Notes | +| --- | --- | --- | --- | +| `yolo11n/` | `r8.im/ultralytics/yolo11n` | `YOLO` | Official YOLO11n object detection model | +| `yolov8s-worldv2/` | `r8.im/ultralytics/yolov8s-worldv2` | `YOLOWorld` | Open-vocabulary YOLOv8s WorldV2 model | +| `yoloe11s/` | `r8.im/ultralytics/yoloe-11s` | `YOLOE` | YOLOE-11S segmentation model with class prompt support | ## ๐Ÿ”ง Model Setup -The model will be automatically downloaded by ultralytics when needed: +Each model directory includes a `download.py` script used by the deployment workflow before `cog build`: -```python -from ultralytics import YOLO - -model = YOLO("yolo11n.pt") # Downloads automatically if not present +```bash +python yolo11n/download.py +python yolov8s-worldv2/download.py +python yoloe11s/download.py ``` ## ๐Ÿงช Local Testing @@ -103,16 +120,17 @@ Test the model locally before deploying: ```bash # Test YOLO11n -python test_prediction.py --model yolo11n --image test.jpg +python yolo11n/download.py +python test_prediction.py --model yolo11n --image assets/bus.jpg ``` ## ๐Ÿš€ Features - **๐ŸŽ๏ธ Optimized**: PyTorch model for fast inference - **๐Ÿค– Automated**: GitHub Actions for CI/CD -- **๐Ÿ“ฆ Ready-to-use**: Pre-configured YOLO11n deployment +- **๐Ÿ“ฆ Ready-to-use**: Pre-configured deployments for multiple YOLO models - **๐Ÿ“Š Scalable**: Auto-scaling Replicate infrastructure -- **๐ŸŽฏ Simple**: Single model focus +- **๐ŸŽฏ Focused**: One Cog configuration per model ## ๐Ÿ’ก Contribute diff --git a/yolo11n/README.md b/yolo11n/README.md index dd73a72..9674160 100644 --- a/yolo11n/README.md +++ b/yolo11n/README.md @@ -23,7 +23,7 @@ Deploy the official YOLO11n model to Replicate with PyTorch inference at https:/ ## Configuration - **GPU**: Disabled by default (CPU inference) -- **Python**: 3.11 with PyTorch 2.0+ +- **Python**: 3.12 with PyTorch 2.0+ - **Framework**: Ultralytics 8.3+ - **Input**: Single image with configurable confidence/IoU thresholds - **Output**: Annotated image with detected objects diff --git a/yoloe11s/README.md b/yoloe11s/README.md index 8347d24..ff7818c 100644 --- a/yoloe11s/README.md +++ b/yoloe11s/README.md @@ -19,7 +19,7 @@ Deploy the official YOLOE-11S model to Replicate with PyTorch inference at Date: Wed, 10 Jun 2026 23:41:06 +0000 Subject: [PATCH 2/3] Auto-format by https://ultralytics.com/actions --- README.md | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 1bb4786..59f9f5a 100644 --- a/README.md +++ b/README.md @@ -98,11 +98,11 @@ pip install -r requirements.txt ## ๐ŸŽฏ Available Models -| Directory | Replicate model | Predictor | Notes | -| --- | --- | --- | --- | -| `yolo11n/` | `r8.im/ultralytics/yolo11n` | `YOLO` | Official YOLO11n object detection model | -| `yolov8s-worldv2/` | `r8.im/ultralytics/yolov8s-worldv2` | `YOLOWorld` | Open-vocabulary YOLOv8s WorldV2 model | -| `yoloe11s/` | `r8.im/ultralytics/yoloe-11s` | `YOLOE` | YOLOE-11S segmentation model with class prompt support | +| Directory | Replicate model | Predictor | Notes | +| ------------------ | ----------------------------------- | ----------- | ------------------------------------------------------ | +| `yolo11n/` | `r8.im/ultralytics/yolo11n` | `YOLO` | Official YOLO11n object detection model | +| `yolov8s-worldv2/` | `r8.im/ultralytics/yolov8s-worldv2` | `YOLOWorld` | Open-vocabulary YOLOv8s WorldV2 model | +| `yoloe11s/` | `r8.im/ultralytics/yoloe-11s` | `YOLOE` | YOLOE-11S segmentation model with class prompt support | ## ๐Ÿ”ง Model Setup From f4fc319a86faddc31afd5d336d8899e95491078f Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 11 Jun 2026 01:46:02 +0200 Subject: [PATCH 3/3] Address README review feedback --- README.md | 3 +++ yoloe11s/README.md | 3 ++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 59f9f5a..abb8a8e 100644 --- a/README.md +++ b/README.md @@ -60,14 +60,17 @@ cd replicate # Deploy YOLO11n cd yolo11n +python download.py cog login cog push r8.im/ultralytics/yolo11n # Or deploy another configured model cd ../yolov8s-worldv2 +python download.py cog push r8.im/ultralytics/yolov8s-worldv2 cd ../yoloe11s +python download.py cog push r8.im/ultralytics/yoloe-11s ``` diff --git a/yoloe11s/README.md b/yoloe11s/README.md index ff7818c..dd59378 100644 --- a/yoloe11s/README.md +++ b/yoloe11s/README.md @@ -19,7 +19,8 @@ Deploy the official YOLOE-11S model to Replicate with PyTorch inference at