trainer.py 798 B

123456789101112131415161718192021222324252627
  1. # !/usr/bin/env python3
  2. # -*- coding: UTF-8 -*-
  3. ################################################################################
  4. #
  5. # Copyright (c) 2024 Baidu.com, Inc. All Rights Reserved
  6. #
  7. ################################################################################
  8. """
  9. Author: PaddlePaddle Authors
  10. """
  11. from ..object_detection import DetTrainer
  12. from .model_list import MODELS
  13. class InstanceSegTrainer(DetTrainer):
  14. """ Instance Segmentation Model Trainer """
  15. entities = MODELS
  16. def _update_dataset(self):
  17. """update dataset settings
  18. """
  19. self.pdx_config.update_dataset(
  20. self.global_config.dataset_dir,
  21. "COCOInstSegDataset",
  22. data_fields=[
  23. 'image', 'gt_bbox', 'gt_class', 'gt_poly', 'is_crowd'
  24. ])