From ef34100280c3032110378232b8d2ac64ac67cb45 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Mon, 23 Sep 2024 11:38:52 +0200
Subject: [PATCH] Removed roughed-in COCO models

---
 model_server/extensions/yolov8/__init__.py |  0
 model_server/extensions/yolov8/models.py   | 43 ----------------------
 2 files changed, 43 deletions(-)
 delete mode 100644 model_server/extensions/yolov8/__init__.py
 delete mode 100644 model_server/extensions/yolov8/models.py

diff --git a/model_server/extensions/yolov8/__init__.py b/model_server/extensions/yolov8/__init__.py
deleted file mode 100644
index e69de29b..00000000
diff --git a/model_server/extensions/yolov8/models.py b/model_server/extensions/yolov8/models.py
deleted file mode 100644
index 1b524054..00000000
--- a/model_server/extensions/yolov8/models.py
+++ /dev/null
@@ -1,43 +0,0 @@
-from typing import List
-
-from pydantic import BaseModel
-from ultralytics import YOLO
-
-from ...base.accessors import GenericImageDataAccessor
-from ...base.models import InstanceSegmentationModel
-from ...base.roiset import RoiSet
-
-class YoloV8Params(BaseModel):
-    pt_file: str
-    duplicate: bool = True
-
-class YoloModel(InstanceSegmentationModel):
-
-    def __init__(self, params: YoloV8Params, autoload=True):
-        # initialize from pretrained model
-        super().__init__(autoload, params)
-
-    def load(self):
-        self.yolo = YOLO(self.params.pt_file)
-
-    def _infer_yolo_seg(self, img:GenericImageDataAccessor) -> RoiSet:
-        # only populates bounding box info
-        pass
-
-    def _infer_yolo_det(self, img:GenericImageDataAccessor) -> RoiSet:
-        # also populates segmentation masks
-        pass
-
-    def label_instance_class(
-            self, img: GenericImageDataAccessor, mask: GenericImageDataAccessor, **kwargs
-    ) -> GenericImageDataAccessor:
-        roiset = self._infer_yolo_seg(img)
-        return roiset.acc_obj_ids
-
-    def export(self):
-        # export pretrained model
-        pass
-
-    def train(self, roisets: List[RoiSet]):
-        coco_list = [r.serialize_coco for r in roisets]
-        self.yolo.train(coco_list)
\ No newline at end of file
-- 
GitLab