mmrotate

mmrotate.models

backbones

dense_heads

detectors

losses

roi_heads

mmrotate.core

mmrotate.datasets

class mmrotate.datasets.DOTADataset(ann_file, pipeline, version='oc', difficulty=100, **kwargs)

DOTA dataset for detection.

Parameters

• ann_file (str) – Annotation file path.

• pipeline (list[dict]) – Processing pipeline.

• version (str, optional) – Angle representations. Defaults to ‘oc’.

• difficulty (bool, optional) – The difficulty threshold of GT.

evaluate(results, metric='mAP', logger=None, proposal_nums=(100, 300, 1000), iou_thr=0.5, scale_ranges=None)

Evaluate the dataset.

Parameters

• results (list) – Testing results of the dataset.

• metric (str | list[str]) – Metrics to be evaluated.

• logger (logging.Logger | None | str) – Logger used for printing related information during evaluation. Default: None.

• proposal_nums (Sequence[int]) – Proposal number used for evaluating recalls, such as recall@100, recall@1000. Default: (100, 300, 1000).

• iou_thr (float | list[float]) – IoU threshold. It must be a float when evaluating mAP, and can be a list when evaluating recall. Default: 0.5.

• scale_ranges (list[tuple] | None) – Scale ranges for evaluating mAP. Default: None.

format_results(results, submission_dir=None, **kwargs)

Format the results to submission text (standard format for DOTA evaluation).

Parameters

• results (list) – Testing results of the dataset.

• submission_dir (str, optional) – The folder that contains submission

• files. – If not specified, a temp folder will be created. Default: None.

Returns

(result_files, tmp_dir), result_files is a dict containing

the json filepaths, tmp_dir is the temporal directory created for saving json files when submission_dir is not specified.

Return type

tuple

load_annotations(ann_folder)

Params:

ann_folder: folder that contains DOTA v1 annotations txt files

merge_det(results)

Merging patch bboxes into full image.

Params:

results (list): Testing results of the dataset.

class mmrotate.datasets.SARDataset(ann_file, pipeline, version='oc', difficulty=100, **kwargs)

SAR ship dataset for detection (Support RSSDD and HRSID).