paddlex 3.0.0b2__py3-none-any.whl → 3.0.0rc1__py3-none-any.whl

paddlex/inference/results/seal_rec.py

@@ -1,50 +0,0 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from pathlib import Path
-from .base import BaseResult, CVResult
-
-
-class SealOCRResult(CVResult):
-    """SealOCRResult"""
-
-    def get_target_name(self, save_path):
-        input_path = self["src_file_name"]
-        if input_path.endswith(".pdf"):
-            save_path = (
-                Path(save_path)
-                / f"{Path(input_path).stem}_pdf"
-                / Path("page_{:04d}".format(self["page_id"] + 1))
-            )
-        else:
-            save_path = Path(save_path) / f"{Path(input_path).stem}"
-        return save_path
-
-    def save_to_img(self, save_path):
-        if not save_path.lower().endswith((".jpg", ".png")):
-            save_path = self.get_target_name(save_path)
-        else:
-            save_path = Path(save_path).stem
-        layout_save_path = f"{save_path}_layout.jpg"
-        layout_result = self["layout_result"]
-        layout_result.save_to_img(layout_save_path)
-        seal_result = self["ocr_result"]
-        seal_result.save_to_img(f"{save_path}_seal_ocr.jpg")
-
-    def save_to_json(self, save_path):
-        if not save_path.lower().endswith((".json")):
-            save_path = self.get_target_name(save_path)
-        else:
-            save_path = Path(save_path).stem
-        super().save_to_json(f"{save_path}_res.json")

paddlex/inference/results/seg.py

@@ -1,72 +0,0 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import numpy as np
-import PIL
-from PIL import Image
-import copy
-import json
-
-from ...utils import logging
-from .utils.mixin import ImgMixin
-from .base import CVResult
-
-
-class SegResult(CVResult):
-    """Save Result Transform"""
-
-    def __init__(self, data):
-        super().__init__(data)
-        self._img_writer.set_backend("pillow", format_="PNG")
-
-    def _to_img(self):
-        """apply"""
-        seg_map = self["pred"]
-        pc_map = self.get_pseudo_color_map(seg_map[0])
-        return pc_map
-
-    def get_pseudo_color_map(self, pred):
-        """get_pseudo_color_map"""
-        if pred.min() < 0 or pred.max() > 255:
-            raise ValueError("`pred` cannot be cast to uint8.")
-        pred = pred.astype(np.uint8)
-        pred_mask = Image.fromarray(pred, mode="P")
-        color_map = self._get_color_map_list(256)
-        pred_mask.putpalette(color_map)
-        return pred_mask
-
-    @staticmethod
-    def _get_color_map_list(num_classes, custom_color=None):
-        """_get_color_map_list"""
-        num_classes += 1
-        color_map = num_classes * [0, 0, 0]
-        for i in range(0, num_classes):
-            j = 0
-            lab = i
-            while lab:
-                color_map[i * 3] |= ((lab >> 0) & 1) << (7 - j)
-                color_map[i * 3 + 1] |= ((lab >> 1) & 1) << (7 - j)
-                color_map[i * 3 + 2] |= ((lab >> 2) & 1) << (7 - j)
-                j += 1
-                lab >>= 3
-        color_map = color_map[3:]
-
-        if custom_color:
-            color_map[: len(custom_color)] = custom_color
-        return color_map
-
-    def _to_str(self, _, *args, **kwargs):
-        data = copy.deepcopy(self)
-        data["pred"] = "..."
-        return super()._to_str(data, *args, **kwargs)

paddlex/inference/results/shitu.py

@@ -1,35 +0,0 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import numpy as np
-from .base import CVResult
-from .det import draw_box
-
-
-class ShiTuResult(CVResult):
-
-    def _to_img(self):
-        """apply"""
-        image = self._img_reader.read(self["input_path"])
-        boxes = [
-            {
-                "coordinate": box["coordinate"],
-                "label": box["labels"][0],
-                "score": box["rec_scores"][0],
-            }
-            for box in self["boxes"]
-            if box["rec_scores"] is not None
-        ]
-        image = draw_box(image, boxes)
-        return image

paddlex/inference/results/table_rec.py

@@ -1,109 +0,0 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import cv2
-import numpy as np
-from pathlib import Path
-import PIL
-from PIL import Image, ImageDraw, ImageFont
-
-from .utils.mixin import HtmlMixin, XlsxMixin
-from .base import BaseResult, CVResult
-
-
-class TableRecResult(CVResult):
-    """SaveTableResults"""
-
-    def __init__(self, data):
-        super().__init__(data)
-
-    def _to_img(self):
-        image = self._img_reader.read(self["input_path"])
-        bbox_res = self["bbox"]
-        if len(bbox_res) > 0 and len(bbox_res[0]) == 4:
-            vis_img = self.draw_rectangle(image, bbox_res)
-        else:
-            vis_img = self.draw_bbox(image, bbox_res)
-        return vis_img
-
-    def draw_rectangle(self, image, boxes):
-        """draw_rectangle"""
-        boxes = np.array(boxes)
-        img_show = image.copy()
-        for box in boxes.astype(int):
-            x1, y1, x2, y2 = box
-            cv2.rectangle(img_show, (x1, y1), (x2, y2), (255, 0, 0), 2)
-        return img_show
-
-    def draw_bbox(self, image, boxes):
-        """draw_bbox"""
-        for box in boxes:
-            box = np.reshape(np.array(box), [-1, 1, 2]).astype(np.int64)
-            image = cv2.polylines(np.array(image), [box], True, (255, 0, 0), 2)
-        return image
-
-
-class StructureTableResult(TableRecResult, HtmlMixin, XlsxMixin):
-    """StructureTableResult"""
-
-    def __init__(self, data):
-        super().__init__(data)
-        HtmlMixin.__init__(self)
-        XlsxMixin.__init__(self)
-
-    def _to_html(self):
-        return self["html"]
-
-
-class TableResult(CVResult, HtmlMixin, XlsxMixin):
-    """TableResult"""
-
-    def __init__(self, data):
-        super().__init__(data)
-        HtmlMixin.__init__(self)
-        XlsxMixin.__init__(self)
-
-    def save_to_html(self, save_path):
-        if not save_path.lower().endswith(("html")):
-            input_path = self["input_path"]
-            save_path = Path(save_path) / f"{Path(input_path).stem}"
-        else:
-            save_path = Path(save_path).stem
-        for table_result in self["table_result"]:
-            table_result.save_to_html(save_path)
-
-    def save_to_xlsx(self, save_path):
-        if not save_path.lower().endswith(("xlsx")):
-            input_path = self["input_path"]
-            save_path = Path(save_path) / f"{Path(input_path).stem}"
-        else:
-            save_path = Path(save_path).stem
-        for table_result in self["table_result"]:
-            table_result.save_to_xlsx(save_path)
-
-    def save_to_img(self, save_path):
-        if not save_path.lower().endswith((".jpg", ".png")):
-            input_path = self["input_path"]
-            save_path = Path(save_path) / f"{Path(input_path).stem}"
-        else:
-            save_path = Path(save_path).stem
-        layout_save_path = f"{save_path}_layout.jpg"
-        ocr_save_path = f"{save_path}_ocr.jpg"
-        table_save_path = f"{save_path}_table"
-        layout_result = self["layout_result"]
-        layout_result.save_to_img(layout_save_path)
-        ocr_result = self["ocr_result"]
-        ocr_result.save_to_img(ocr_save_path)
-        for idx, table_result in enumerate(self["table_result"]):
-            table_result.save_to_img(f"{table_save_path}_{idx}.jpg")

paddlex/inference/results/text_det.py

@@ -1,33 +0,0 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import numpy as np
-import cv2
-
-from .base import CVResult
-
-
-class TextDetResult(CVResult):
-    def __init__(self, data):
-        super().__init__(data)
-        self._img_reader.set_backend("opencv")
-
-    def _to_img(self):
-        """draw rectangle"""
-        boxes = self["dt_polys"]
-        image = self._img_reader.read(self["input_path"])
-        for box in boxes:
-            box = np.reshape(np.array(box).astype(int), [-1, 1, 2]).astype(np.int64)
-            cv2.polylines(image, [box], True, (0, 0, 255), 2)
-        return image

paddlex/inference/results/text_rec.py

@@ -1,66 +0,0 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import PIL
-from PIL import Image, ImageDraw, ImageFont
-import numpy as np
-import cv2
-
-from ...utils.fonts import PINGFANG_FONT_FILE_PATH
-from ...utils import logging
-from .base import CVResult
-
-
-class TextRecResult(CVResult):
-    def _to_img(self):
-        """Draw label on image"""
-        image = self._img_reader.read(self["input_path"])
-        rec_text = self["rec_text"]
-        rec_score = self["rec_score"]
-        image = image.convert("RGB")
-        image_width, image_height = image.size
-        text = f"{rec_text} ({rec_score})"
-        font = self.adjust_font_size(image_width, text, PINGFANG_FONT_FILE_PATH)
-        row_height = font.getbbox(text)[3]
-        new_image_height = image_height + int(row_height * 1.2)
-        new_image = Image.new("RGB", (image_width, new_image_height), (255, 255, 255))
-        new_image.paste(image, (0, 0))
-
-        draw = ImageDraw.Draw(new_image)
-        draw.text(
-            (0, image_height),
-            text,
-            fill=(0, 0, 0),
-            font=font,
-        )
-        return new_image
-
-    def adjust_font_size(self, image_width, text, font_path):
-        font_size = int(image_width * 0.06)
-        font = ImageFont.truetype(font_path, font_size)
-
-        if int(PIL.__version__.split(".")[0]) < 10:
-            text_width, _ = font.getsize(text)
-        else:
-            text_width, _ = font.getbbox(text)[2:]
-
-        while text_width > image_width:
-            font_size -= 1
-            font = ImageFont.truetype(font_path, font_size)
-            if int(PIL.__version__.split(".")[0]) < 10:
-                text_width, _ = font.getsize(text)
-            else:
-                text_width, _ = font.getbbox(text)[2:]
-
-        return font

paddlex/inference/results/ts.py

@@ -1,37 +0,0 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from .utils.mixin import JsonMixin, CSVMixin
-from .base import BaseResult
-
-
-class _BaseTSResult(BaseResult, CSVMixin):
-    def __init__(self, data):
-        super().__init__(data)
-        CSVMixin.__init__(self)
-
-
-class TSFcResult(_BaseTSResult):
-    def _to_csv(self):
-        return self["forecast"]
-
-
-class TSClsResult(_BaseTSResult):
-    def _to_csv(self):
-        return self["classification"]
-
-
-class TSAdResult(_BaseTSResult):
-    def _to_csv(self):
-        return self["anomaly"]

paddlex/inference/results/utils/__init__.py

@@ -1,13 +0,0 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.

paddlex/inference/results/utils/mixin.py

@@ -1,204 +0,0 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from abc import abstractmethod
-import json
-from pathlib import Path
-import numpy as np
-from PIL import Image
-import pandas as pd
-
-from ....utils import logging
-from ...utils.io import (
-    JsonWriter,
-    ImageReader,
-    ImageWriter,
-    CSVWriter,
-    HtmlWriter,
-    XlsxWriter,
-    TextWriter,
-)
-
-
-def _save_list_data(save_func, save_path, data, *args, **kwargs):
-    save_path = Path(save_path)
-    if data is None:
-        return
-    if isinstance(data, list):
-        for idx, single in enumerate(data):
-            save_func(
-                (
-                    save_path.parent / f"{save_path.stem}_{idx}{save_path.suffix}"
-                ).as_posix(),
-                single,
-                *args,
-                **kwargs,
-            )
-    save_func(save_path.as_posix(), data, *args, **kwargs)
-    logging.info(f"The result has been saved in {save_path}.")
-
-
-class StrMixin:
-    @property
-    def str(self):
-        return self._to_str()
-
-    def _to_str(self, data, json_format=False, indent=4, ensure_ascii=False):
-        if json_format:
-            return json.dumps(data.json, indent=indent, ensure_ascii=ensure_ascii)
-        else:
-            return str(data)
-
-    def print(self, json_format=False, indent=4, ensure_ascii=False):
-        str_ = self._to_str(
-            self, json_format=json_format, indent=indent, ensure_ascii=ensure_ascii
-        )
-        logging.info(str_)
-
-
-class JsonMixin:
-    def __init__(self):
-        self._json_writer = JsonWriter()
-        self._show_funcs.append(self.save_to_json)
-
-    def _to_json(self):
-        def _format_data(obj):
-            if isinstance(obj, np.float32):
-                return float(obj)
-            elif isinstance(obj, np.ndarray):
-                return [_format_data(item) for item in obj.tolist()]
-            elif isinstance(obj, pd.DataFrame):
-                return json.loads(obj.to_json(orient="records", force_ascii=False))
-            elif isinstance(obj, Path):
-                return obj.as_posix()
-            elif isinstance(obj, dict):
-                return type(obj)({k: _format_data(v) for k, v in obj.items()})
-            elif isinstance(obj, (list, tuple)):
-                return [_format_data(i) for i in obj]
-            else:
-                return obj
-
-        return _format_data(self)
-
-    @property
-    def json(self):
-        return self._to_json()
-
-    def save_to_json(self, save_path, indent=4, ensure_ascii=False, *args, **kwargs):
-        if not str(save_path).endswith(".json"):
-            save_path = Path(save_path) / f"{Path(self['input_path']).stem}.json"
-        _save_list_data(
-            self._json_writer.write,
-            save_path,
-            self.json,
-            indent=indent,
-            ensure_ascii=ensure_ascii,
-            *args,
-            **kwargs,
-        )
-
-
-class Base64Mixin:
-    def __init__(self, *args, **kwargs):
-        self._base64_writer = TextWriter(*args, **kwargs)
-        self._show_funcs.append(self.save_to_base64)
-
-    @abstractmethod
-    def _to_base64(self):
-        raise NotImplementedError
-
-    @property
-    def base64(self):
-        return self._to_base64()
-
-    def save_to_base64(self, save_path, *args, **kwargs):
-        if not str(save_path).lower().endswith((".b64")):
-            fp = Path(self["input_path"])
-            save_path = Path(save_path) / f"{fp.stem}{fp.suffix}"
-        _save_list_data(
-            self._base64_writer.write, save_path, self.base64, *args, **kwargs
-        )
-
-
-class ImgMixin:
-    def __init__(self, backend="pillow", *args, **kwargs):
-        self._img_writer = ImageWriter(backend=backend, *args, **kwargs)
-        self._show_funcs.append(self.save_to_img)
-
-    @abstractmethod
-    def _to_img(self):
-        raise NotImplementedError
-
-    @property
-    def img(self):
-        image = self._to_img()
-        # The img must be a PIL.Image obj
-        if isinstance(image, np.ndarray):
-            return Image.fromarray(image)
-        return image
-
-    def save_to_img(self, save_path, *args, **kwargs):
-        if not str(save_path).lower().endswith((".jpg", ".png")):
-            fp = Path(self["input_path"])
-            save_path = Path(save_path) / f"{fp.stem}{fp.suffix}"
-        _save_list_data(self._img_writer.write, save_path, self.img, *args, **kwargs)
-
-
-class CSVMixin:
-    def __init__(self, backend="pandas", *args, **kwargs):
-        self._csv_writer = CSVWriter(backend=backend, *args, **kwargs)
-        self._show_funcs.append(self.save_to_csv)
-
-    @abstractmethod
-    def _to_csv(self):
-        raise NotImplementedError
-
-    def save_to_csv(self, save_path, *args, **kwargs):
-        if not str(save_path).endswith(".csv"):
-            save_path = Path(save_path) / f"{Path(self['input_path']).stem}.csv"
-        _save_list_data(
-            self._csv_writer.write, save_path, self._to_csv(), *args, **kwargs
-        )
-
-
-class HtmlMixin:
-    def __init__(self, *args, **kwargs):
-        self._html_writer = HtmlWriter(*args, **kwargs)
-        self._show_funcs.append(self.save_to_html)
-
-    @property
-    def html(self):
-        return self._to_html()
-
-    def _to_html(self):
-        return self["html"]
-
-    def save_to_html(self, save_path, *args, **kwargs):
-        if not str(save_path).endswith(".html"):
-            save_path = Path(save_path) / f"{Path(self['input_path']).stem}.html"
-        _save_list_data(self._html_writer.write, save_path, self.html, *args, **kwargs)
-
-
-class XlsxMixin:
-    def __init__(self, *args, **kwargs):
-        self._xlsx_writer = XlsxWriter(*args, **kwargs)
-        self._show_funcs.append(self.save_to_xlsx)
-
-    def _to_xlsx(self):
-        return self["html"]
-
-    def save_to_xlsx(self, save_path, *args, **kwargs):
-        if not str(save_path).endswith(".xlsx"):
-            save_path = Path(save_path) / f"{Path(self['input_path']).stem}.xlsx"
-        _save_list_data(self._xlsx_writer.write, save_path, self.html, *args, **kwargs)

paddlex/inference/results/warp.py

@@ -1,31 +0,0 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import numpy as np
-import copy
-import json
-
-from ...utils import logging
-from .base import CVResult
-
-
-class DocTrResult(CVResult):
-
-    def _to_img(self):
-        return np.array(self["doctr_img"])
-
-    def _to_str(self, _, *args, **kwargs):
-        data = copy.deepcopy(self)
-        data.pop("doctr_img")
-        return super()._to_str(data, *args, **kwargs)

paddlex/inference/utils/new_ir_blacklist.py

@@ -1,22 +0,0 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-NEWIR_BLOCKLIST = [
-    "PP-YOLOE_seg-S",
-    "PatchTST_ad",
-    "TimesNet_ad",
-    "Nonstationary_ad",
-    "DLinear_ad",
-    "LaTeX_OCR_rec",
-]