memolib 1.0.0__tar.gz

memolib-1.0.0/MemoLib/Model/BaseModel/IModel.py

@@ -0,0 +1,25 @@
+from abc import ABC, abstractmethod
+from .eModelBase import eModelExportType
+
+class IModel(ABC):
+    @abstractmethod
+    def LoadWeight(self, weightPath: str): pass
+
+    @abstractmethod
+    def Predict(self, img): pass
+
+    @abstractmethod
+    def BatchPredict(self, batchImg): pass
+
+    @abstractmethod
+    def Train(self): pass
+
+    @abstractmethod
+    def StopTraining(self): pass
+
+    @abstractmethod
+    def Export(self, pathToPytorchModel: str, exportType: eModelExportType): pass
+
+    @abstractmethod
+    def StopTraining(self): pass
+

memolib-1.0.0/MemoLib/Model/BaseModel/__init__.py

@@ -0,0 +1,25 @@
+from .IModel import IModel
+from .eModelBase import eModelType, eDetectionModelType, eSegmentationModelType
+from .eModelBase import eClassificationModelType, eRecognitionModelType
+from .eModelBase import eAnomalyDetectionModelType
+from .eModelBase import eModelExportType, eOptimizerType
+from .eDetectionModel import eYoloDetectionModel, eRFDetrDetectionModel
+from .eSegmentationModel import eYoloSegmentationModel
+from .eClassificationModel import (
+    eYoloClassificationModel, eRFDetrClassificationModel
+)
+from .eClassificationModel import eEfficientNetModel, ePPLCNetModel
+from .eAnomalyModel import eDinomalyModel, eINPFormerlyModel
+
+__all__ = [
+    'IModel',
+    'eModelType', 'eDetectionModelType', 'eSegmentationModelType',
+    'eClassificationModelType', 'eRecognitionModelType',
+    'eAnomalyDetectionModelType',
+    'eModelExportType', 'eOptimizerType',
+    'eYoloDetectionModel', 'eRFDetrDetectionModel',
+    'eYoloSegmentationModel',
+    'eYoloClassificationModel', 'eRFDetrClassificationModel',
+    'eEfficientNetModel', 'ePPLCNetModel',
+    'eDinomalyModel', 'eINPFormerlyModel',
+]

memolib-1.0.0/MemoLib/Model/BaseModel/eAnomalyModel.py

@@ -0,0 +1,12 @@
+from MemoLib.Model.BaseModel.eModelBase import eAnomalyDetectionModelType
+
+class eDinomalyModel(eAnomalyDetectionModelType):
+    Dinomaly_S = 0
+    Dinomaly_B = 1
+    Dinomaly_L = 2
+
+class eINPFormerlyModel(eAnomalyDetectionModelType):
+    INFormerly_S = 0
+    INFormerly_B = 1
+    INFormerly_L = 2
+

memolib-1.0.0/MemoLib/Model/BaseModel/eClassificationModel.py

@@ -0,0 +1,52 @@
+from MemoLib.Model.BaseModel.eModelBase import eClassificationModelType
+
+class eYoloClassificationModel(eClassificationModelType):
+    YoloV5n = 0
+    YoloV5s = 1
+    YoloV5m = 2
+    YoloV5l = 3
+    YoloV5x = 4
+    YoloV8n = 5
+    YoloV8s = 6
+    YoloV8m = 7
+    YoloV8l = 8
+    YoloV8x = 9
+    Yolo11n = 10
+    Yolo11s = 11
+    Yolo11m = 12
+    Yolo11l = 13
+    Yolo11x = 14
+    Yolo26n = 15
+    Yolo26s = 16
+    Yolo26m = 17
+    Yolo26l = 18
+    Yolo26x = 19
+
+class eRFDetrClassificationModel(eClassificationModelType):
+    RFDETRNano = 0
+    RFDETRSmall = 1
+    RFDETRMedium = 2
+    RFDETRLarge = 3
+    RFDETRXLarge = 4
+    RFDETR2XLarge = 5
+    
+class eEfficientNetModel(eClassificationModelType):
+    B0 = 0
+    B1 = 1
+    B2 = 2
+    B3 = 3
+    B4 = 4
+    B5 = 5
+    B6 = 6
+    B7 = 7
+    V2S = 8
+    V2M = 9
+    V2L = 10
+    V2XL = 11
+
+class ePPLCNetModel(eClassificationModelType):
+    PPLCNetx25 = 0
+    PPLCNetx35 = 1
+    PPLCNetx50 = 2
+    PPLCNetx75 = 3
+    PPLCNetx100 = 4

memolib-1.0.0/MemoLib/Model/BaseModel/eDetectionModel.py

@@ -0,0 +1,34 @@
+
+from MemoLib.Model.BaseModel.eModelBase import eDetectionModelType
+
+class eYoloDetectionModel(eDetectionModelType):
+    YoloV5n = 0
+    YoloV5s = 1
+    YoloV5m = 2
+    YoloV5l = 3
+    YoloV5x = 4
+    YoloV8n = 5
+    YoloV8s = 6
+    YoloV8m = 7
+    YoloV8l = 8
+    YoloV8x = 9
+    Yolo11n = 10
+    Yolo11s = 11
+    Yolo11m = 12
+    Yolo11l = 13
+    Yolo11x = 14
+    Yolo26n = 15
+    Yolo26s = 16
+    Yolo26m = 17
+    Yolo26l = 18
+    Yolo26x = 19
+
+class eRFDetrDetectionModel(eDetectionModelType):
+    RFDETRNano = 0
+    RFDETRSmall = 1
+    RFDETRMedium = 2
+    RFDETRLarge = 3
+    RFDETRXLarge = 4
+    RFDETR2XLarge = 5
+
+

memolib-1.0.0/MemoLib/Model/BaseModel/eModelBase.py

@@ -0,0 +1,31 @@
+from enum import Enum
+
+class eModelType(Enum):
+    pass
+
+class eDetectionModelType(eModelType):
+    pass
+
+class eSegmentationModelType(eModelType):
+    pass
+
+class eClassificationModelType(eModelType):
+    pass
+
+class eRecognitionModelType(eModelType):
+    pass
+
+class eAnomalyDetectionModelType(eModelType):
+    pass
+
+class eModelExportType(Enum):
+    ONNX = 0
+    OpenVINO = 1
+    TensorRT = 2
+
+class eOptimizerType(Enum):
+    SGD = 0
+    Adam = 1
+    AdamW = 2
+    RMSprop = 3
+    Adagrad = 4

memolib-1.0.0/MemoLib/Model/BaseModel/eSegmentationModel.py

@@ -0,0 +1,36 @@
+from .eModelBase import eSegmentationModelType
+
+class eYoloSegmentationModel(eSegmentationModelType):
+    YoloV5n = 0
+    YoloV5s = 1
+    YoloV5m = 2
+    YoloV5l = 3
+    YoloV5x = 4
+    YoloV8n = 5
+    YoloV8s = 6
+    YoloV8m = 7
+    YoloV8l = 8
+    YoloV8x = 9
+    Yolo11n = 10
+    Yolo11s = 11
+    Yolo11m = 12
+    Yolo11l = 13
+    Yolo11x = 14
+    Yolo26n = 15
+    Yolo26s = 16
+    Yolo26m = 17
+    Yolo26l = 18
+    Yolo26x = 19
+
+class eDinoUperNetModel(eSegmentationModelType):
+    DINO_S = 0
+    DINO_B = 1
+    DINO_L = 2
+
+class eRFDetrSegmentationModel(eSegmentationModelType):
+    RFDETRNano = 0
+    RFDETRSmall = 1
+    RFDETRMedium = 2
+    RFDETRLarge = 3
+    RFDETRXLarge = 4
+    RFDETR2XLarge = 5