# !/usr/bin/env python3 # -*- coding: UTF-8 -*- ################################################################################ # # Copyright (c) 2024 Baidu.com, Inc. All Rights Reserved # ################################################################################ """ Author: PaddlePaddle Authors """ import os from .dataset_src import check, convert, split_dataset, deep_analyse from ...base import BaseDatasetChecker from ..model_list import MODELS class COCOInstSegDatasetChecker(BaseDatasetChecker): """Dataset Checker for Instance Segmentation Model """ entities = MODELS sample_num = 10 def convert_dataset(self, src_dataset_dir: str) -> str: """convert the dataset from other type to specified type Args: src_dataset_dir (str): the root directory of dataset. Returns: str: the root directory of converted dataset. """ return convert(self.check_dataset_config.convert.src_dataset_type, src_dataset_dir) def split_dataset(self, src_dataset_dir: str) -> str: """repartition the train and validation dataset Args: src_dataset_dir (str): the root directory of dataset. Returns: str: the root directory of splited dataset. """ return split_dataset(src_dataset_dir, self.check_dataset_config.split_train_percent, self.check_dataset_config.split_val_percent) def check_dataset(self, dataset_dir: str, sample_num: int=sample_num) -> dict: """check if the dataset meets the specifications and get dataset summary Args: dataset_dir (str): the root directory of dataset. sample_num (int): the number to be sampled. Returns: dict: dataset summary. """ return check(dataset_dir, self.output) def analyse(self, dataset_dir: str) -> dict: """deep analyse dataset Args: dataset_dir (str): the root directory of dataset. Returns: dict: the deep analysis results. """ return deep_analyse(dataset_dir, self.output) def get_show_type(self) -> str: """get the show type of dataset Returns: str: show type """ return "image" def get_dataset_type(self) -> str: """return the dataset type Returns: str: dataset type """ return "COCOInstSegDataset"