gpclipper 0.0.2__tar.gz

gpclipper-0.0.2/PKG-INFO

@@ -0,0 +1,31 @@
+Metadata-Version: 2.4
+Name: gpclipper
+Version: 0.0.2
+Summary: Garment clipping (matting) pipeline — GP2Clipper
+Author: sanchi
+Author-email: sanchi@flixstock.com
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.8
+Requires-Dist: torch
+Requires-Dist: torchvision
+Requires-Dist: numpy
+Requires-Dist: opencv-python
+Requires-Dist: Pillow
+Requires-Dist: termcolor
+Requires-Dist: scikit-learn
+Requires-Dist: pandas
+Requires-Dist: tqdm
+Requires-Dist: ipdb
+Provides-Extra: train
+Requires-Dist: tensorboardX; extra == "train"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+Garment clipping (matting) pipeline — GP2Clipper

gpclipper-0.0.2/gpclipper/MlModules/BaseConfig.py

@@ -0,0 +1,6 @@
+import os
+
+class BaseConfig:
+    def __init__(self):
+        self.base_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
+        self.checkpoints_weights_path = os.path.join(self.base_path, "Checkpoints-n-Weights")

gpclipper-0.0.2/gpclipper/MlModules/BaseModelClass.py

@@ -0,0 +1,49 @@
+import os
+import random
+import time
+import sys
+import ssl
+import shutil
+import uuid
+from abc import ABC, abstractmethod
+
+from ..helpler.imageHelper import ImageHelper
+from ..helpler.localHelper import LocalContent
+from ..helpler.zipHelper import ZipHelper
+
+
+class BaseModel(ABC):
+    def __init__(self, init_api=True, checkpoints_dir=None):
+        self.base_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
+        sys.argv = [sys.argv[0]]
+        ssl._create_default_https_context = ssl._create_unverified_context
+        if checkpoints_dir is not None:
+            self.checkpoints_weights_path = os.path.realpath(checkpoints_dir)
+        else:
+            self.checkpoints_weights_path = os.path.join(self.base_path, "Checkpoints-n-Weights")
+        self.env_name = "clipper18_V2"
+        self.temp_base_path = os.path.join(self.base_path, "temp")
+        os.makedirs(self.temp_base_path, exist_ok=True)
+        self.LocalContent = LocalContent
+        self.ImageHelper = ImageHelper
+        self.ZipHelper = ZipHelper
+
+    def get_unique_path(self):
+        current_time = str(time.time())
+        random_value = random.randint(11111, 99999)
+        path = os.path.join(self.temp_base_path, current_time + str(random_value))
+        if os.path.exists(path):
+            return self.get_unique_path()
+        os.makedirs(path)
+        return path
+
+    def getUniqueFilename(self, ext):
+        return f"{uuid.uuid4()}{ext}"
+
+    @abstractmethod
+    def load_model(self):
+        pass
+
+    @abstractmethod
+    def run_model(self):
+        pass