142 lines
4.0 KiB
Python
142 lines
4.0 KiB
Python
from app.service.design_fast.pipeline import LoadImage, KeyPoint, Segmentation, Color, PrintPainting, Scaling, Split, LoadBodyImage, ContourDetection, NoSegPrintPainting
|
|
|
|
|
|
class BaseItem:
|
|
def __init__(self, data, basic):
|
|
self.result = data.copy()
|
|
self.result['name'] = data['type'].lower()
|
|
self.result.pop("type")
|
|
self.result.update(basic)
|
|
self.result['design_type'] = basic.get('design_type', None)
|
|
|
|
|
|
class OthersItem(BaseItem):
|
|
def __init__(self, data, basic, minio_client):
|
|
super().__init__(data, basic)
|
|
self.Others_pipeline = [
|
|
LoadImage(minio_client),
|
|
Segmentation(minio_client),
|
|
Scaling(),
|
|
Split(minio_client)
|
|
]
|
|
|
|
def process(self):
|
|
for item in self.Others_pipeline:
|
|
self.result = item(self.result)
|
|
return self.result
|
|
|
|
|
|
class TopItem(BaseItem):
|
|
def __init__(self, data, basic, minio_client):
|
|
super().__init__(data, basic)
|
|
self.top_pipeline = [
|
|
LoadImage(minio_client),
|
|
KeyPoint(),
|
|
Segmentation(minio_client),
|
|
# BackPerspective(minio_client),
|
|
Color(minio_client),
|
|
NoSegPrintPainting(minio_client),
|
|
PrintPainting(minio_client),
|
|
Scaling(),
|
|
Split(minio_client)
|
|
]
|
|
|
|
def process(self):
|
|
for item in self.top_pipeline:
|
|
self.result = item(self.result)
|
|
return self.result
|
|
|
|
|
|
class BottomItem(BaseItem):
|
|
def __init__(self, data, basic, minio_client):
|
|
super().__init__(data, basic)
|
|
self.bottom_pipeline = [
|
|
LoadImage(minio_client),
|
|
KeyPoint(),
|
|
ContourDetection(),
|
|
Segmentation(minio_client),
|
|
# BackPerspective(minio_client),
|
|
Color(minio_client),
|
|
NoSegPrintPainting(minio_client),
|
|
PrintPainting(minio_client),
|
|
Scaling(),
|
|
Split(minio_client)
|
|
]
|
|
|
|
def process(self):
|
|
for item in self.bottom_pipeline:
|
|
self.result = item(self.result)
|
|
return self.result
|
|
|
|
|
|
"""merge"""
|
|
|
|
|
|
class OthersMergeItem(BaseItem):
|
|
def __init__(self, data, basic, minio_client):
|
|
super().__init__(data, basic)
|
|
self.Others_pipeline = [
|
|
LoadImage(minio_client),
|
|
# KeyPoint(),
|
|
# ContourDetection(),
|
|
Segmentation(minio_client),
|
|
# BackPerspective(minio_client),
|
|
Color(minio_client),
|
|
NoSegPrintPainting(minio_client),
|
|
PrintPainting(minio_client),
|
|
Scaling(),
|
|
Split(minio_client)
|
|
]
|
|
|
|
def process(self):
|
|
for item in self.Others_pipeline:
|
|
self.result = item(self.result)
|
|
return self.result
|
|
|
|
|
|
class TopMergeItem(BaseItem):
|
|
def __init__(self, data, basic, minio_client):
|
|
super().__init__(data, basic)
|
|
self.top_pipeline = [
|
|
LoadImage(minio_client),
|
|
KeyPoint(),
|
|
Segmentation(minio_client),
|
|
Scaling(),
|
|
Split(minio_client)
|
|
]
|
|
|
|
def process(self):
|
|
for item in self.top_pipeline:
|
|
self.result = item(self.result)
|
|
return self.result
|
|
|
|
|
|
class BottomMergeItem(BaseItem):
|
|
def __init__(self, data, basic, minio_client):
|
|
super().__init__(data, basic)
|
|
self.bottom_pipeline = [
|
|
LoadImage(minio_client),
|
|
KeyPoint(),
|
|
Segmentation(minio_client),
|
|
Scaling(),
|
|
Split(minio_client)
|
|
]
|
|
|
|
def process(self):
|
|
for item in self.bottom_pipeline:
|
|
self.result = item(self.result)
|
|
return self.result
|
|
|
|
|
|
class BodyItem(BaseItem):
|
|
def __init__(self, data, basic, minio_client):
|
|
super().__init__(data, basic)
|
|
self.top_pipeline = [
|
|
LoadBodyImage(minio_client),
|
|
]
|
|
|
|
def process(self):
|
|
for item in self.top_pipeline:
|
|
self.result = item(self.result)
|
|
return self.result
|