| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687 |
- # !/usr/bin/env python3
- # -*- coding: UTF-8 -*-
- ################################################################################
- #
- # Copyright (c) 2024 Baidu.com, Inc. All Rights Reserved
- #
- ################################################################################
- """
- Author: PaddlePaddle Authors
- """
- import os
- import os.path as osp
- from ...base import BaseDatasetChecker
- from .dataset_src import check_dataset, convert_dataset, split_dataset, anaylse_dataset
- from ..model_list import MODELS
- class SegDatasetChecker(BaseDatasetChecker):
- """ Dataset Checker for Semantic 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_dataset(self.check_dataset_config.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(dataset_dir, self.output, sample_num)
- 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 anaylse_dataset(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 "SegDataset"
|