Plugin: Predict plugin based on YOLO8

.github/workflows/plugins-docker.yaml

@@ -4,19 +4,35 @@ on:
       - "ext/**"
 
 jobs:
-  publish-hello-docker-image:
+  publish-plugin-docker-images:
+    strategy:
+      fail-fast: false
+      matrix:
+        plugins:
+          - realesrgan
+          - yolo8
+
     runs-on: ubuntu-latest
     steps:
+      - uses: dorny/paths-filter@v2
+        id: changes
+        with:
+          filters: |
+            changed: 'ext/${{ matrix.plugins }}/**'
+
       - name: Checkout
+        if: steps.changes.outputs.changed == 'true'
         uses: actions/checkout@v3
 
       - name: Log in to Docker Hub
+        if: steps.changes.outputs.changed == 'true'
         uses: docker/login-action@v1
         with:
           username: ${{ secrets.DOCKER_USERNAME }}
           password: ${{ secrets.DOCKER_PASSWORD }}
 
       - name: Build and push Docker image
+        if: steps.changes.outputs.changed == 'true'
         uses: docker/build-push-action@v2
         with:
           context: ext/realesrgan

ext/yolo8/Dockerfile

@@ -0,0 +1,13 @@
+FROM python:3.10.11-slim-buster
+
+RUN apt-get update && apt-get install -y libgl1 libglib2.0-0
+
+COPY requirements.txt /app/
+RUN pip install -r /app/requirements.txt
+
+COPY . /app/
+WORKDIR /app
+
+EXPOSE 5000
+
+CMD ["python3", "main.py"]

ext/yolo8/main.py

@@ -0,0 +1,26 @@
+import base64
+from flask import Flask, request
+from io import BytesIO
+from PIL import Image
+from ultralytics import YOLO
+from ultralytics.yolo.engine.results import Results
+
+app = Flask(__name__)
+model = YOLO("yolov8n.pt")
+
+@app.route('/hello', methods=['GET'])
+def hello():
+    return "elloh"
+
+@app.route('/predict', methods=['POST'])
+def predict():
+    image_data = base64.b64decode(request.json['image'])
+    image = Image.open(BytesIO(image_data))
+
+    results: Results = model.predict(image)
+
+    return results[0].tojson()
+
+if __name__ == '__main__':
+    print("running")
+    app.run(host='0.0.0.0', port=5000)

ext/yolo8/requirements.txt

@@ -0,0 +1,2 @@
+Flask
+ultralytics