From ca587f84ed51883f3553a9ee0cc7ed8405b62110 Mon Sep 17 00:00:00 2001
From: wuyefeilin <30919197+wuyefeilin@users.noreply.github.com>
Date: Wed, 2 Aug 2023 10:34:01 +0800
Subject: [PATCH] add dataset alias for PaddleX (#8502)

---
 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