o 1ju@sLddlZddlmZddlmZddlmZddlmZGdd d eZ dS) N)Union)logging)abspath) BaseSegConfigc@seZdZdZddedefddZdefdd Zd eee ffd d Z d efddZ d e fddZ dd efddZded efddZdS) SegConfigzSemantic Segmentation ConfigN dataset_path dataset_typec Cstt|}|dur d}|dkr||}||dS|dkr1|dd|tj|dddidStt|d) zupdate dataset settings Args: dataset_path (str): the root path of dataset. dataset_type (str, optional): dataset type. Defaults to None. Raises: ValueError: the dataset_type error. N SegDatasetZ_dummy val_datasetval.txtvaltype dataset_rootZval_pathmodez is not supported.)r_make_custom_dataset_configupdateospathjoin ValueErrorrepr)selfr r Z dataset_dirds_cfgrk/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/repo_apis/PaddleSeg_api/seg/config.pyupdate_datasets"    zSegConfig.update_dataset num_classescCsd|vr ||jd<d|vr||jd<d|vr||jd<|jdvr@|jdD] }|ddkr1||d<q%|jd }|D]}||d<q9d Sd S) zmupdate classes number Args: num_classes (int): the classes number value to set. train_datasetrr model)ZMaskFormer_tinyZMaskFormer_small transformsrZGenerateInstanceTargetstypesN)r r r!Z model_nameZloss)rrtf_cfgZlossesZloss_cfgrrrupdate_num_classes;s        zSegConfig.update_num_classes crop_sizecCst|tr ||g}nt|}t|dkrtt|dt|dg}|jd}d}|D]}|ddkr9||d<d }q+|sCtd d Sd S) zupdate the image cropping size of training preprocessing Args: crop_size (int | list): the size of image to be cropped. Raises: ValueError: the `crop_size` error. rrr"FrZRandomPaddingCropr&TzvCould not find configuration item of image cropping transformation operator. Therefore, the crop size was not updated.N) isinstanceintlistlenrr rwarning)rr&Z tf_cfg_listmodifiedr$rrrupdate_train_crop_sizeQs$    z SegConfig.update_train_crop_sizereturncCsd|vr|jSdS)ziget epochs Returns: int: the epochs value, i.e., `Global.epochs` in config. itersi)r0rrrrget_epochs_itersoszSegConfig.get_epochs_iterscCs d|vs d|jvr dS|jdS)zget learning rate Returns: float: the learning rate value, i.e., `Optimizer.lr.learning_rate` in config. lr_schedulerZ learning_rateg-C6?)r3r1rrrget_learning_rate{s zSegConfig.get_learning_ratetraincCs.|dkr d|vr |jSdStdt|d)aget batch size Args: mode (str, optional): the mode that to be get batch size value, must be one of 'train', 'eval', 'test'. Defaults to 'train'. Raises: ValueError: the `mode` error. `train` is supported only. Returns: int: the batch size value of `mode`, i.e., `DataLoader.{mode}.sampler.batch_size` in config. r5 batch_sizerzGetting `batch_size` in z mode is not supported.)r6rr)rrrrrget_batch_sizes zSegConfig.get_batch_sizedataset_root_pathcCs6d|tj|dddd|tj|dddd}|S) zconstruct the dataset config that meets the format requirements Args: dataset_root_path (str): the root directory of dataset. Returns: dict: the dataset config. r z train.txtr5)rrZ train_pathrr rr)r r )rrr)rr8rrrrrs   z%SegConfig._make_custom_dataset_config)N)r5)__name__ __module__ __qualname____doc__strrr)r%rr*r.r2floatr4r7dictrrrrrrs   r) rtypingrutilsrZ utils.miscrZbase_seg_configrrrrrrs