init datamate

runtime/ops/mapper/img_resize/__init__.py

@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+
+from datamate.core.base_op import OPERATORS
+
+OPERATORS.register_module(module_name='ImgResize',
+                          module_path="ops.mapper.img_resize.process")

runtime/ops/mapper/img_resize/metadata.yml

@@ -0,0 +1,35 @@
+name: '图片重采样'
+name_en: 'Image Resampling'
+description: '将图片放大或缩小到指定像素。'
+description_en: 'Zooms in or out images to specified pixels.'
+language: 'python'
+vendor: 'huawei'
+raw_id: 'ImgResize'
+version: '1.0.0'
+types:
+  - 'cleanse'
+modal: 'image'
+effect:
+  before: ''
+  after: ''
+inputs: 'image'
+outputs: 'image'
+settings:
+  targetSize:
+    name: 重采样尺寸
+    type: multiple
+    properties:
+      - type: inputNumber
+        name: 宽度
+        description: 像素
+        defaultVal: 256
+        min: 1
+        max: 4096
+        step: 1
+      - type: inputNumber
+        name: 高度
+        description: 像素
+        defaultVal: 256
+        min: 1
+        max: 4096
+        step: 1

runtime/ops/mapper/img_resize/process.py

@@ -0,0 +1,40 @@
+# -- encoding: utf-8 --
+
+"""
+Description:
+Create: 2025/01/16
+"""
+import time
+from typing import List, Dict, Any
+
+from loguru import logger
+import cv2
+
+from datamate.common.utils import bytes_transform
+from datamate.core.base_op import Mapper
+
+
+class ImgResize(Mapper):
+    def __init__(self, *args, **kwargs):
+        super(ImgResize, self).__init__(*args, **kwargs)
+        self._target_size = kwargs.get("targetSize", [256, 256])
+
+    @classmethod
+    def _img_resize(cls, data: List[float], target_size: List[int]) -> List[float]:
+        """将图片缩放到指定尺寸大小"""
+        target_width = max(min(target_size[0], 4096), 1)
+        target_height = max(min(target_size[1], 4096), 1)
+        resized_img = cv2.resize(data, (target_width, target_height), interpolation=cv2.INTER_AREA)
+        return resized_img
+
+    def execute(self, sample: Dict[str, Any]):
+        start = time.time()
+        file_name = sample[self.filename_key]
+        file_type = "." + sample[self.filetype_key]
+        img_bytes = sample[self.data_key]
+        if img_bytes:
+            data = bytes_transform.bytes_to_numpy(img_bytes)
+            resized_img = self._img_resize(data, self._target_size)
+            sample[self.data_key] = bytes_transform.numpy_to_bytes(resized_img, file_type)
+            logger.info(f"fileName: {file_name}, method: ImgResize costs {time.time() - start:6f} s")
+        return sample