Supported tasks

For each of the supported tasks, the template includes reference configuration files and implementation of common stages (data, explore, experiment).

Classification

Using mmclassification.

See also: ai-dataset: Classification

Configs

• dataset
• model

Stages

Data

Transform:

• property_split
• random_split
• coco_to_mmclassification

Explore:

• explore_ground_truth_classification
• paint_ground_truth_classification

Model

Explore:

• process_image_classification

---

Detection

Using mmdetection.

See also: ai-dataset: Detection

Configs

• dataset
• model

Stages

Data

Transform:

• property_split
• random_split
• extract_crops_from_detection_bounding_boxes

Explore:

• explore_ground_truth_detection
• paint_ground_truth_detection

Model

Explore:

• process_image_detection

---

Semantic Segmentation

Using mmsegmentation.

See also: ai-dataset: Segmentation

Configs

• dataset
• model

Stages

Data

Transform:

• property_split

• random_split

• coco_to_mmsegmentation

Explore:

• explore_ground_truth_segmentation
• paint_ground_truth_segmentation

Model

Explore:

• process_image_segmentation