| 123456789101112131415161718192021222324252627 |
- # !/usr/bin/env python3
- # -*- coding: UTF-8 -*-
- ################################################################################
- #
- # Copyright (c) 2024 Baidu.com, Inc. All Rights Reserved
- #
- ################################################################################
- """
- Author: PaddlePaddle Authors
- """
- from ..object_detection import DetTrainer
- from .model_list import MODELS
- class InstanceSegTrainer(DetTrainer):
- """ Instance Segmentation Model Trainer """
- entities = MODELS
- def _update_dataset(self):
- """update dataset settings
- """
- self.pdx_config.update_dataset(
- self.global_config.dataset_dir,
- "COCOInstSegDataset",
- data_fields=[
- 'image', 'gt_bbox', 'gt_class', 'gt_poly', 'is_crowd'
- ])
|