未验证 提交 249de9bd 编写于 作者: W wuyefeilin 提交者: GitHub

add dataset alias for PaddleX (#8503)

* add dataset alias for PaddleX

* Update coco.py

---------
Co-authored-by: Nshiyutang <34859558+shiyutang@users.noreply.github.com>
上级 fcea7fdd
......@@ -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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册