From 249de9bdc0411c431292e8d0b603e7583bcd82e4 Mon Sep 17 00:00:00 2001
From: wuyefeilin <30919197+wuyefeilin@users.noreply.github.com>
Date: Tue, 29 Aug 2023 14:24:39 +0800
Subject: [PATCH] add dataset alias for PaddleX (#8503)

* add dataset alias for PaddleX

* Update coco.py

---------

Co-authored-by: shiyutang <34859558+shiyutang@users.noreply.github.com>
---
 ppdet/data/source/coco.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/ppdet/data/source/coco.py b/ppdet/data/source/coco.py
index 330dae677..412032791 100644
--- a/ppdet/data/source/coco.py
+++ b/ppdet/data/source/coco.py
@@ -25,7 +25,9 @@ from .dataset import DetDataset
 from ppdet.utils.logger import setup_logger
 logger = setup_logger(__name__)
 
-__all__ = ['COCODataSet', 'SlicedCOCODataSet', 'SemiCOCODataSet']
+__all__ = [
+    'COCODataSet', 'SlicedCOCODataSet', 'SemiCOCODataSet', 'COCODetDataset'
+]
 
 
 @register
@@ -585,3 +587,10 @@ class SemiCOCODataSet(COCODataSet):
         self._curr_iter += 1
 
         return self.transform(roidb)
+
+
+# for PaddleX
+@register
+@serializable
+class COCODetDataset(COCODataSet):
+    pass
-- 
GitLab