__init__.py 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. # copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
  2. #
  3. # Licensed under the Apache License, Version 2.0 (the "License");
  4. # you may not use this file except in compliance with the License.
  5. # You may obtain a copy of the License at
  6. #
  7. # http://www.apache.org/licenses/LICENSE-2.0
  8. #
  9. # Unless required by applicable law or agreed to in writing, software
  10. # distributed under the License is distributed on an "AS IS" BASIS,
  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. # See the License for the specific language governing permissions and
  13. # limitations under the License.
  14. import os
  15. import os.path as osp
  16. from collections import defaultdict, Counter
  17. from PIL import Image
  18. import json
  19. from ...base import BaseDatasetChecker
  20. from .dataset_src import check, split_dataset, deep_analyse
  21. from ..support_models import SUPPORT_MODELS
  22. class TableRecDatasetChecker(BaseDatasetChecker):
  23. """Dataset Checker for Table Recognition Model
  24. """
  25. support_models = SUPPORT_MODELS
  26. def convert_dataset(self, src_dataset_dir: str) -> str:
  27. """convert the dataset from other type to specified type
  28. Args:
  29. src_dataset_dir (str): the root directory of dataset.
  30. Returns:
  31. str: the root directory of converted dataset.
  32. """
  33. return src_dataset_dir
  34. def split_dataset(self, src_dataset_dir: str) -> str:
  35. """repartition the train and validation dataset
  36. Args:
  37. src_dataset_dir (str): the root directory of dataset.
  38. Returns:
  39. str: the root directory of splited dataset.
  40. """
  41. return split_dataset(src_dataset_dir,
  42. self.check_dataset_config.split.train_percent,
  43. self.check_dataset_config.split.val_percent)
  44. def check_dataset(self, dataset_dir: str) -> dict:
  45. """check if the dataset meets the specifications and get dataset summary
  46. Args:
  47. dataset_dir (str): the root directory of dataset.
  48. Returns:
  49. dict: dataset summary.
  50. """
  51. return check(dataset_dir, self.global_config.output, sample_num=10)
  52. def get_show_type(self) -> str:
  53. """get the show type of dataset
  54. Returns:
  55. str: show type
  56. """
  57. return "image"
  58. def get_dataset_type(self) -> str:
  59. """return the dataset type
  60. Returns:
  61. str: dataset type
  62. """
  63. return "PubTabTableRecDataset"