init

lib/task/__init__.py

@@ -0,0 +1,3 @@
+from .segmentation import SegmentationTask
+
+__all__ = ['SegmentationTask']

lib/task/segmentation.py

@@ -0,0 +1,18 @@
+# coding: utf-8
+# Copyright (c) Ant Group. All rights reserved.
+
+from antmmf.common.registry import registry
+from antmmf.tasks import BaseTask
+
+
+@registry.register_task("segmentation")
+class SegmentationTask(BaseTask):
+
+    def __init__(self):
+        super(SegmentationTask, self).__init__("segmentation")
+
+    def _get_available_datasets(self):
+        return ["pretraining_loader"]
+
+    def _preprocess_item(self, item):
+        return item