oriented-det 0.1.0__py3-none-any.whl

oriented_det/configs/rotated_faster_rcnn/dota_le90_1x.json

@@ -0,0 +1,119 @@
+{
+  "_base_": [
+    "../_base_/datasets/dota_le90.json",
+    "../_base_/models/rotated_faster_rcnn_r50.json",
+    "../_base_/schedules/1x.json",
+    "../_base_/preprocessing.json"
+  ],
+  "enable_albumentation": false,
+  "data_loader": {
+    "batch_size": 2,
+    "num_workers": 4,
+    "shuffle": true,
+    "pin_memory": true
+  },
+  "checkpoint": {
+    "load_from_checkpoint": null,
+    "load_from_experiment": null,
+    "load_include_prefixes": null,
+    "load_exclude_prefixes": null,
+    "discover_previous_run": false,
+    "resume_from_checkpoint_epoch": false,
+    "load_optimizer_state": false,
+    "load_scheduler_state": false,
+    "start_epoch": 0,
+    "best_metric": "mAP",
+    "higher_is_better": true
+  },
+  "training": {
+    "num_epochs": 12,
+    "learning_rate": 0.005,
+    "momentum": 0.9,
+    "weight_decay": 0.0001,
+    "use_amp": false,
+    "gradient_accumulation_steps": 1,
+    "max_grad_norm": 35.0,
+    "lr_scheduler_milestones": [8, 11],
+    "lr_scheduler_gamma": 0.1,
+    "lr_warmup_steps": 500,
+    "lr_scaling_with_accumulation": "linear",
+    "lr_scale_with_world_size": false,
+    "use_lr_param_groups": false
+  },
+  "loss": {
+    "loss_type": "cross_entropy",
+    "label_smoothing": 0.0
+  },
+  "dataset": {
+    "train_tiles_dirs": [
+      "/path/to/data/DOTA-v1.0-tiled/train",
+      "/path/to/data/DOTA-v1.0-tiled/val"
+    ],
+    "val_tiles_dirs": [
+      "/path/to/data/DOTA-v1.0-tiled/val"
+    ],
+    "filter_empty_gt": true,
+    "overlap": 200,
+    "difficult_strategy": "keep",
+    "ignore_labels": [
+      "container-crane",
+      "airport",
+      "helipad"
+    ]
+  },
+  "preprocessing": {
+    "enable_flip_diagonal": true
+  },
+  "evaluation": {
+    "score_threshold": 0.05,
+    "iou_threshold": 0.5,
+    "compute_map_final": false,
+    "compute_map_every_n_epochs": 4,
+    "extended_gt_metrics": true
+  },
+  "tensorboard": {
+    "log_debug_anchors_proposals": true,
+    "vis_score_threshold": 0.5
+  },
+  "model": {
+    "pretrained_backbone": true,
+    "frozen_stages": 1,
+    "inference_pre_nms_score_threshold": 0.05,
+    "rpn_pre_nms_top_n": 2000,
+    "rpn_post_nms_top_n": 2000,
+    "rpn_min_size": 0,
+    "rpn_nms_threshold": 0.7,
+    "final_nms_iou_threshold": 0.3,
+    "roi_box_reg_angle_weight": 1.0,
+    "roi_box_reg_iou_weight": 0.1,
+    "roi_box_reg_iou_loss_type": "probiou",
+    "roi_box_reg_probiou_mode": "l1",
+    "final_nms_use_cpu": true,
+    "nms_class_agnostic": false,
+    "max_detections_per_image": 2000,
+    "rpn_positive_iou_threshold": 0.7,
+    "rpn_negative_iou_threshold": 0.3,
+    "rpn_min_pos_iou": 0.3,
+    "rpn_match_low_quality": true,
+    "roi_positive_iou_threshold": 0.5,
+    "roi_negative_iou_threshold": 0.5,
+    "roi_match_low_quality": false,
+    "roi_batch_size_per_image": 512,
+    "add_gt_as_proposals": true,
+    "roi_inference_top_class_only": true
+  },
+  "production": {
+    "score_threshold": 0.05,
+    "inference_pre_nms_score_threshold": 0.05,
+    "final_nms_iou_threshold": 0.1,
+    "rpn_pre_nms_top_n": 6000,
+    "rpn_post_nms_top_n": 6000,
+    "max_detections_per_image": 2000,
+    "roi_inference_top_class_only": true,
+    "overlap_pixels": 200,
+    "ignore_margin_pixels": 0,
+    "use_first_image_canvas": false,
+    "stick_to_model_canvas": true,
+    "final_nms_use_cpu": true
+  }
+}

oriented_det/configs/rotated_faster_rcnn/dota_le90_3x.json

@@ -0,0 +1,9 @@
+{
+  "_base_": ["./dota_le90_1x.json"],
+  "training": {
+    "num_epochs": 36,
+    "learning_rate": 0.005,
+    "lr_scheduler_milestones": [24, 33],
+    "lr_scheduler_gamma": 0.1
+  }
+}

oriented_det/configs/rotated_retinanet/dota_le90_1x.json

@@ -0,0 +1,107 @@
+{
+  "_base_": [
+    "../_base_/models/rotated_retinanet_r50.json",
+    "../_base_/schedules/1x.json",
+    "../_base_/preprocessing.json"
+  ],
+  "data_loader": {
+    "batch_size": 2,
+    "num_workers": 4,
+    "shuffle": true,
+    "pin_memory": true
+  },
+  "checkpoint": {
+    "load_from_checkpoint": null,
+    "load_from_experiment": null,
+    "discover_previous_run": false,
+    "resume_from_checkpoint_epoch": false,
+    "load_optimizer_state": false,
+    "load_scheduler_state": false,
+    "start_epoch": 0,
+    "best_metric": "mAP",
+    "higher_is_better": true
+  },
+  "training": {
+    "num_epochs": 12,
+    "learning_rate": 0.0025,
+    "momentum": 0.9,
+    "weight_decay": 0.0001,
+    "use_amp": false,
+    "gradient_accumulation_steps": 1,
+    "max_grad_norm": 35.0,
+    "lr_scheduler_milestones": [8, 11],
+    "lr_scheduler_gamma": 0.1,
+    "lr_warmup_steps": 500,
+    "lr_scaling_with_accumulation": "linear",
+    "lr_scale_with_world_size": false,
+    "use_lr_param_groups": false
+  },
+  "loss": {
+    "loss_type": "focal",
+    "focal_alpha": 0.25,
+    "focal_gamma": 2.0
+  },
+  "dataset": {
+    "data_root": "/path/to/data/DOTA-v1.0-tiled",
+    "train_tiles_dirs": [
+      "/path/to/data/DOTA-v1.0-tiled/train",
+      "/path/to/data/DOTA-v1.0-tiled/val"
+    ],
+    "val_tiles_dirs": [
+      "/path/to/data/DOTA-v1.0-tiled/val"
+    ],
+    "overlap": 200,
+    "difficult_strategy": "keep",
+    "filter_empty_gt": true,
+    "ignore_labels": [
+      "container-crane",
+      "airport",
+      "helipad"
+    ]
+  },
+  "preprocessing": {
+    "enable_flip_diagonal": true
+  },
+  "evaluation": {
+    "score_threshold": 0.05,
+    "iou_threshold": 0.5,
+    "compute_map_final": false,
+    "compute_map_every_n_epochs": 4,
+    "extended_gt_metrics": true,
+    "use_exact_rotated_iou": false
+  },
+  "tensorboard": {
+    "log_debug_anchors_proposals": true,
+    "vis_score_threshold": 0.5
+  },
+  "model": {
+    "pretrained_backbone": true,
+    "frozen_stages": 1,
+    "fpn_returned_layers": [2, 3, 4],
+    "fpn_strides": [8, 16, 32, 64, 128],
+    "fpn_extra_level": true,
+    "anchor_octave_base_scale": 4,
+    "anchor_scales_per_octave": 3,
+    "anchor_ratios": [0.5, 1.0, 2.0],
+    "retinanet_stacked_convs": 4,
+    "box_reg_loss_type": "l1",
+    "use_hbb_for_matching": true,
+    "final_nms_use_cpu": false,
+    "inference_pre_nms_score_threshold": 0.05,
+    "final_nms_iou_threshold": 0.1,
+    "max_detections_per_image": 2000,
+    "rpn_positive_iou_threshold": 0.5,
+    "rpn_negative_iou_threshold": 0.4
+  },
+  "production": {
+    "score_threshold": 0.05,
+    "inference_pre_nms_score_threshold": 0.05,
+    "final_nms_iou_threshold": 0.5,
+    "max_detections_per_image": 3000,
+    "overlap_pixels": 200,
+    "ignore_margin_pixels": 0,
+    "use_first_image_canvas": false,
+    "stick_to_model_canvas": true,
+    "final_nms_use_cpu": true
+  }
+}

oriented_det/configs/rotated_retinanet/dota_le90_3x.json

@@ -0,0 +1,30 @@
+{
+  "_base_": ["./dota_le90_1x.json"],
+  "training": {
+    "num_epochs": 36,
+    "lr_scheduler_milestones": [24, 33],
+    "lr_warmup_steps": 500
+  },
+  "evaluation": {
+    "score_threshold": 0.3,
+    "compute_map_final": true,
+    "compute_map_every_n_epochs": 4,
+    "use_exact_rotated_iou": false,
+    "use_exact_rotated_iou_for_final_map": true
+  },
+  "production": {
+    "score_threshold": null
+  },
+  "model": {
+    "max_detections_per_image": 300
+  },
+  "checkpoint": {
+    "load_from_checkpoint": null,
+    "load_from_experiment": null,
+    "discover_previous_run": false,
+    "resume_from_checkpoint_epoch": false,
+    "load_optimizer_state": false,
+    "load_scheduler_state": false,
+    "start_epoch": 0
+  }
+}

oriented_det/data/__init__.py

@@ -0,0 +1,89 @@
+"""Data loading, tiling, transforms, and evaluation for oriented detection."""
+
+from .dota import (
+    DOTAAnnotation,
+    DOTADataset,
+    DOTASample,
+    build_dota_loader,
+    build_dota_split_dataset,
+    collect_dota_image_paths,
+    collect_dota_split_image_paths,
+    dota_dataset_class_names,
+    dota_label_path_for_image,
+    format_dota_empty_gt_filter_log,
+    format_dota_line,
+    resolve_dota_tile_roots,
+)
+from .airbus_playground import (
+    AirbusPlaygroundCSVDataset,
+    detect_airbus_split_csv_format,
+    format_airbus_empty_gt_filter_log,
+    generate_airbus_playground_csvs,
+)
+from .evaluation import (
+    APCalculator,
+    ClassEvalMetrics,
+    Detection,
+    GroundTruth,
+    compute_oriented_map,
+    format_mmrotate_class_metrics_table,
+)
+from .tiling import (
+    ImageTiler,
+    Tile,
+    TiledSample,
+    visualize_tiles,
+)
+from .flips import apply_random_train_flips
+from .transforms import (
+    AlbumentationsTransform,
+    Compose,
+    DiagonalFlip,
+    HorizontalFlip,
+    OrientedTransform,
+    Rotate,
+    VerticalFlip,
+    create_albumentations_augmentation,
+)
+
+__all__ = [
+    # DOTA dataset
+    "DOTAAnnotation",
+    "DOTADataset",
+    "DOTASample",
+    "build_dota_loader",
+    "build_dota_split_dataset",
+    "collect_dota_image_paths",
+    "collect_dota_split_image_paths",
+    "dota_dataset_class_names",
+    "dota_label_path_for_image",
+    "format_dota_empty_gt_filter_log",
+    "format_dota_line",
+    "resolve_dota_tile_roots",
+    "AirbusPlaygroundCSVDataset",
+    "detect_airbus_split_csv_format",
+    "format_airbus_empty_gt_filter_log",
+    "generate_airbus_playground_csvs",
+    # Tiling
+    "ImageTiler",
+    "Tile",
+    "TiledSample",
+    "visualize_tiles",
+    # Transforms
+    "AlbumentationsTransform",
+    "Compose",
+    "create_albumentations_augmentation",
+    "apply_random_train_flips",
+    "HorizontalFlip",
+    "DiagonalFlip",
+    "OrientedTransform",
+    "Rotate",
+    "VerticalFlip",
+    # Evaluation
+    "APCalculator",
+    "ClassEvalMetrics",
+    "Detection",
+    "GroundTruth",
+    "compute_oriented_map",
+    "format_mmrotate_class_metrics_table",
+]