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

add dataset alias for PaddleX (#8502)

上级 03f8464d
...@@ -25,7 +25,9 @@ from .dataset import DetDataset ...@@ -25,7 +25,9 @@ from .dataset import DetDataset
from ppdet.utils.logger import setup_logger from ppdet.utils.logger import setup_logger
logger = setup_logger(__name__) logger = setup_logger(__name__)
__all__ = ['COCODataSet', 'SlicedCOCODataSet', 'SemiCOCODataSet'] __all__ = [
'COCODataSet', 'SlicedCOCODataSet', 'SemiCOCODataSet', 'COCODetDataset'
]
@register @register
...@@ -585,3 +587,10 @@ class SemiCOCODataSet(COCODataSet): ...@@ -585,3 +587,10 @@ class SemiCOCODataSet(COCODataSet):
self._curr_iter += 1 self._curr_iter += 1
return self.transform(roidb) 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.
先完成此消息的编辑!
想要评论请 注册