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

add dataset alias for PaddleX (#8502)

上级 03f8464d
......@@ -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.
先完成此消息的编辑!
想要评论请 注册