oriented-det 0.1.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- export/__init__.py +9 -0
- export/ort_runtime.py +67 -0
- export/postprocess.py +151 -0
- export/scripts/__init__.py +6 -0
- export/scripts/build_faster_rcnn_savedmodel.py +104 -0
- export/scripts/export_onnx.py +210 -0
- export/scripts/onnx_to_savedmodel.py +35 -0
- export/scripts/predict_savedmodel.py +94 -0
- export/scripts/save_predictions_tf.py +447 -0
- export/scripts/to_tflite.py +32 -0
- export/tests/test_export_onnx_optional.py +82 -0
- export/tests/test_export_wrappers.py +105 -0
- export/tests/test_faster_rcnn_export_parity.py +201 -0
- export/tests/test_ort_runtime.py +41 -0
- export/tf_serving_model.py +96 -0
- export/val_dataset.py +116 -0
- export/wrappers.py +161 -0
- oriented_det/__init__.py +77 -0
- oriented_det/cli/__init__.py +92 -0
- oriented_det/cli/train.py +18 -0
- oriented_det/configs/_base_/augmentation.json +21 -0
- oriented_det/configs/_base_/datasets/dota_le90.json +21 -0
- oriented_det/configs/_base_/fp16.json +5 -0
- oriented_det/configs/_base_/models/oriented_rcnn_r50.json +26 -0
- oriented_det/configs/_base_/models/rotated_faster_rcnn_r50.json +53 -0
- oriented_det/configs/_base_/models/rotated_retinanet_r50.json +30 -0
- oriented_det/configs/_base_/preprocessing.json +8 -0
- oriented_det/configs/_base_/schedules/1x.json +33 -0
- oriented_det/configs/config.schema.json +404 -0
- oriented_det/configs/oriented_rcnn/dota_le90_1x.json +121 -0
- oriented_det/configs/oriented_rcnn/dota_le90_3x.json +11 -0
- oriented_det/configs/rotated_faster_rcnn/dota_le90_1x.json +119 -0
- oriented_det/configs/rotated_faster_rcnn/dota_le90_3x.json +9 -0
- oriented_det/configs/rotated_retinanet/dota_le90_1x.json +107 -0
- oriented_det/configs/rotated_retinanet/dota_le90_3x.json +30 -0
- oriented_det/data/__init__.py +89 -0
- oriented_det/data/airbus_playground.py +611 -0
- oriented_det/data/dota.py +741 -0
- oriented_det/data/dota_classes.py +26 -0
- oriented_det/data/evaluation.py +648 -0
- oriented_det/data/flips.py +115 -0
- oriented_det/data/preprocessing.py +335 -0
- oriented_det/data/tiling.py +399 -0
- oriented_det/data/transforms.py +377 -0
- oriented_det/geometry/__init__.py +8 -0
- oriented_det/geometry/poly.py +127 -0
- oriented_det/geometry/qbox.py +63 -0
- oriented_det/geometry/rbox.py +250 -0
- oriented_det/geometry/transforms.py +266 -0
- oriented_det/models/__init__.py +36 -0
- oriented_det/models/backbones/__init__.py +11 -0
- oriented_det/models/backbones/resnet_fpn.py +79 -0
- oriented_det/models/backbones/utils.py +81 -0
- oriented_det/models/bbox_coder.py +355 -0
- oriented_det/models/faster_rcnn_inference.py +494 -0
- oriented_det/models/horizontal_roi_coder.py +155 -0
- oriented_det/models/oriented_rcnn.py +1256 -0
- oriented_det/models/oriented_roi.py +1664 -0
- oriented_det/models/oriented_rpn.py +2104 -0
- oriented_det/models/rotated_retinanet.py +1030 -0
- oriented_det/models/utils.py +590 -0
- oriented_det/ops/__init__.py +58 -0
- oriented_det/ops/gpu_ops.py +1109 -0
- oriented_det/ops/iou.py +172 -0
- oriented_det/ops/kfiou.py +275 -0
- oriented_det/ops/nms.py +202 -0
- oriented_det/ops/probiou.py +165 -0
- oriented_det/ops/rotated_ops.py +122 -0
- oriented_det/ops/utils.py +257 -0
- oriented_det/pretrained/__init__.py +23 -0
- oriented_det/pretrained/hub.py +249 -0
- oriented_det/pretrained/manifest.json +46 -0
- oriented_det/runtime/__init__.py +29 -0
- oriented_det/runtime/checkpoint.py +274 -0
- oriented_det/runtime/collate.py +348 -0
- oriented_det/runtime/inference.py +1286 -0
- oriented_det/train/__init__.py +102 -0
- oriented_det/train/config.py +872 -0
- oriented_det/train/engine.py +2933 -0
- oriented_det/train/grouped_ce.py +139 -0
- oriented_det/train/piecewise_schedule.py +33 -0
- oriented_det/train/profiler.py +287 -0
- oriented_det/train/utils.py +999 -0
- oriented_det/utils/__init__.py +31 -0
- oriented_det/utils/config.py +376 -0
- oriented_det/utils/device.py +35 -0
- oriented_det/utils/logging.py +163 -0
- oriented_det/utils/progress.py +62 -0
- oriented_det/utils/viz.py +181 -0
- oriented_det-0.1.0.dist-info/METADATA +313 -0
- oriented_det-0.1.0.dist-info/RECORD +115 -0
- oriented_det-0.1.0.dist-info/WHEEL +5 -0
- oriented_det-0.1.0.dist-info/entry_points.txt +2 -0
- oriented_det-0.1.0.dist-info/licenses/LICENSE +202 -0
- oriented_det-0.1.0.dist-info/top_level.txt +3 -0
- tools/__init__.py +1 -0
- tools/app.py +1284 -0
- tools/dataset_stats.py +389 -0
- tools/dota_labels_to_comma.py +132 -0
- tools/free_gpu.py +142 -0
- tools/generate_airbus_playground_csv.py +154 -0
- tools/image_demo.py +200 -0
- tools/lr_finder.py +771 -0
- tools/measure_sampled_riou_error.py +483 -0
- tools/playground_to_dota.py +290 -0
- tools/pretrained_download.py +49 -0
- tools/preview_augmentation.py +510 -0
- tools/publish_checkpoint.py +96 -0
- tools/save_predictions.py +2350 -0
- tools/sync_vendored_configs.py +94 -0
- tools/tile_dota.py +447 -0
- tools/train.py +2324 -0
- tools/train_example.py +244 -0
- tools/train_multi_gpu.py +367 -0
- tools/visualize_boxes.py +183 -0
|
@@ -0,0 +1,181 @@
|
|
|
1
|
+
"""Visualization helpers for geometric primitives."""
|
|
2
|
+
|
|
3
|
+
from __future__ import annotations
|
|
4
|
+
|
|
5
|
+
from dataclasses import dataclass
|
|
6
|
+
from typing import Iterable, List, Sequence, Tuple
|
|
7
|
+
import itertools
|
|
8
|
+
import math
|
|
9
|
+
import random
|
|
10
|
+
|
|
11
|
+
try: # Optional dependency; only required when drawing actual images.
|
|
12
|
+
from PIL import Image, ImageDraw, ImageFont
|
|
13
|
+
except Exception: # pragma: no cover - exercised when pillow is not installed.
|
|
14
|
+
Image = None # type: ignore[assignment]
|
|
15
|
+
ImageDraw = None # type: ignore[assignment]
|
|
16
|
+
ImageFont = None # type: ignore[assignment]
|
|
17
|
+
|
|
18
|
+
from ..geometry import Polygon, QBox, RBox
|
|
19
|
+
|
|
20
|
+
Point = Tuple[float, float]
|
|
21
|
+
Color = Tuple[int, int, int]
|
|
22
|
+
|
|
23
|
+
DEFAULT_PALETTE = [
|
|
24
|
+
(253, 128, 93),
|
|
25
|
+
(44, 160, 44),
|
|
26
|
+
(31, 119, 180),
|
|
27
|
+
(148, 103, 189),
|
|
28
|
+
(214, 39, 40),
|
|
29
|
+
(140, 86, 75),
|
|
30
|
+
(188, 189, 34),
|
|
31
|
+
(23, 190, 207),
|
|
32
|
+
]
|
|
33
|
+
|
|
34
|
+
|
|
35
|
+
def as_polygon(obj) -> Polygon:
|
|
36
|
+
if isinstance(obj, Polygon):
|
|
37
|
+
return obj
|
|
38
|
+
if isinstance(obj, QBox):
|
|
39
|
+
return obj.to_polygon()
|
|
40
|
+
if isinstance(obj, RBox):
|
|
41
|
+
return obj.to_polygon()
|
|
42
|
+
return Polygon(obj)
|
|
43
|
+
|
|
44
|
+
|
|
45
|
+
def cycle_palette(n: int, *, palette: Sequence[Color] | None = None) -> List[Color]:
|
|
46
|
+
palette = palette or DEFAULT_PALETTE
|
|
47
|
+
if not palette:
|
|
48
|
+
raise ValueError("Palette cannot be empty.")
|
|
49
|
+
repeated = list(itertools.islice(itertools.cycle(palette), n))
|
|
50
|
+
return repeated
|
|
51
|
+
|
|
52
|
+
|
|
53
|
+
def random_palette(n: int, seed: int | None = None) -> List[Color]:
|
|
54
|
+
rng = random.Random(seed)
|
|
55
|
+
return [(rng.randint(0, 255), rng.randint(0, 255), rng.randint(0, 255)) for _ in range(n)]
|
|
56
|
+
|
|
57
|
+
|
|
58
|
+
def format_label(label: str, score: float | None = None) -> str:
|
|
59
|
+
return f"{label} ({score:.2f})" if score is not None else label
|
|
60
|
+
|
|
61
|
+
|
|
62
|
+
@dataclass
|
|
63
|
+
class DrawingSpec:
|
|
64
|
+
"""Parameters controlling drawing appearance."""
|
|
65
|
+
|
|
66
|
+
outline: Color = (255, 255, 255)
|
|
67
|
+
fill: Color | None = None
|
|
68
|
+
width: int = 2
|
|
69
|
+
|
|
70
|
+
|
|
71
|
+
def _ensure_pil_image(image):
|
|
72
|
+
if Image is None or ImageDraw is None:
|
|
73
|
+
raise RuntimeError("Pillow is required for drawing visualizations.")
|
|
74
|
+
if isinstance(image, Image.Image):
|
|
75
|
+
return image.convert("RGB")
|
|
76
|
+
try:
|
|
77
|
+
import numpy as np
|
|
78
|
+
|
|
79
|
+
if isinstance(image, np.ndarray):
|
|
80
|
+
return Image.fromarray(image.astype("uint8"), "RGB")
|
|
81
|
+
except Exception: # pragma: no cover - optional dependency
|
|
82
|
+
pass
|
|
83
|
+
raise TypeError("image must be a PIL.Image or a numpy array.")
|
|
84
|
+
|
|
85
|
+
|
|
86
|
+
def _polygon_centroid(poly_pts: Sequence[Sequence[float]]) -> Tuple[float, float]:
|
|
87
|
+
"""Return (cx, cy) centroid of polygon points."""
|
|
88
|
+
pts = [tuple(map(float, pt)) for pt in poly_pts]
|
|
89
|
+
if not pts:
|
|
90
|
+
return (0.0, 0.0)
|
|
91
|
+
n = len(pts)
|
|
92
|
+
cx = sum(p[0] for p in pts) / n
|
|
93
|
+
cy = sum(p[1] for p in pts) / n
|
|
94
|
+
return (cx, cy)
|
|
95
|
+
|
|
96
|
+
|
|
97
|
+
def _polygon_top_anchor(poly_pts: Sequence[Sequence[float]], offset: float = 2.0) -> Tuple[float, float]:
|
|
98
|
+
"""Return (x, y) position for label at the top of the polygon (smallest y = top in image coords).
|
|
99
|
+
Uses the midpoint of the top edge so the label sits near the top without obscuring the box content.
|
|
100
|
+
"""
|
|
101
|
+
pts = [tuple(map(float, pt)) for pt in poly_pts]
|
|
102
|
+
if not pts:
|
|
103
|
+
return (0.0, 0.0)
|
|
104
|
+
sorted_by_y = sorted(pts, key=lambda p: p[1])
|
|
105
|
+
top_two = sorted_by_y[:2]
|
|
106
|
+
mx = (top_two[0][0] + top_two[1][0]) / 2
|
|
107
|
+
my = (top_two[0][1] + top_two[1][1]) / 2
|
|
108
|
+
return (mx, my - offset)
|
|
109
|
+
|
|
110
|
+
|
|
111
|
+
def draw_polygons(
|
|
112
|
+
image,
|
|
113
|
+
polygons: Iterable[Sequence[Sequence[float]]],
|
|
114
|
+
specs: Sequence[DrawingSpec] | None = None,
|
|
115
|
+
labels: Iterable[str] | None = None,
|
|
116
|
+
label_color: Color = (255, 255, 255),
|
|
117
|
+
):
|
|
118
|
+
"""Draw polygons on image, optionally with text labels near the top of each polygon.
|
|
119
|
+
|
|
120
|
+
Args:
|
|
121
|
+
image: PIL Image or numpy array (RGB).
|
|
122
|
+
polygons: Iterable of polygon point lists.
|
|
123
|
+
specs: Optional drawing specs (outline, fill, width) per polygon.
|
|
124
|
+
labels: Optional list of text labels; if provided, length must match polygons.
|
|
125
|
+
label_color: RGB color for label text (default white).
|
|
126
|
+
|
|
127
|
+
Returns:
|
|
128
|
+
PIL Image with polygons and labels drawn.
|
|
129
|
+
"""
|
|
130
|
+
pil_img = _ensure_pil_image(image)
|
|
131
|
+
draw = ImageDraw.Draw(pil_img, "RGBA")
|
|
132
|
+
polygons = list(polygons)
|
|
133
|
+
specs = specs or [DrawingSpec(outline=color) for color in cycle_palette(len(polygons) or 1)]
|
|
134
|
+
if len(specs) != len(polygons):
|
|
135
|
+
raise ValueError("specs length must match polygons length.")
|
|
136
|
+
labels_list = list(labels) if labels is not None else None
|
|
137
|
+
if labels_list is not None and len(labels_list) != len(polygons):
|
|
138
|
+
raise ValueError("labels length must match polygons length when provided.")
|
|
139
|
+
font = None
|
|
140
|
+
if ImageFont is not None:
|
|
141
|
+
for path in (
|
|
142
|
+
"/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf",
|
|
143
|
+
"/usr/share/fonts/truetype/liberation/LiberationSans-Regular.ttf",
|
|
144
|
+
):
|
|
145
|
+
try:
|
|
146
|
+
font = ImageFont.truetype(path, 10)
|
|
147
|
+
break
|
|
148
|
+
except Exception:
|
|
149
|
+
continue
|
|
150
|
+
if font is None:
|
|
151
|
+
try:
|
|
152
|
+
font = ImageFont.load_default()
|
|
153
|
+
except Exception:
|
|
154
|
+
pass
|
|
155
|
+
for i, (poly_pts, spec) in enumerate(zip(polygons, specs)):
|
|
156
|
+
pts = [tuple(map(float, pt)) for pt in poly_pts]
|
|
157
|
+
if spec.fill:
|
|
158
|
+
draw.polygon(pts, fill=spec.fill + (64,))
|
|
159
|
+
draw.line(pts + [pts[0]], fill=spec.outline, width=spec.width)
|
|
160
|
+
if labels_list is not None and i < len(labels_list) and labels_list[i]:
|
|
161
|
+
x, y = _polygon_top_anchor(poly_pts)
|
|
162
|
+
kwargs = {"fill": label_color}
|
|
163
|
+
if font is not None:
|
|
164
|
+
kwargs["font"] = font
|
|
165
|
+
draw.text((x, y), labels_list[i], **kwargs)
|
|
166
|
+
return pil_img
|
|
167
|
+
|
|
168
|
+
|
|
169
|
+
def draw_boxes(image, boxes: Iterable[RBox | QBox | Sequence[float]], **kwargs):
|
|
170
|
+
polygons = [as_polygon(box).points for box in boxes]
|
|
171
|
+
return draw_polygons(image, polygons, **kwargs)
|
|
172
|
+
|
|
173
|
+
|
|
174
|
+
__all__ = [
|
|
175
|
+
"cycle_palette",
|
|
176
|
+
"random_palette",
|
|
177
|
+
"format_label",
|
|
178
|
+
"DrawingSpec",
|
|
179
|
+
"draw_polygons",
|
|
180
|
+
"draw_boxes",
|
|
181
|
+
]
|
|
@@ -0,0 +1,313 @@
|
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
|
+
Name: oriented-det
|
|
3
|
+
Version: 0.1.0
|
|
4
|
+
Summary: A lightweight, modern PyTorch library for rotated object detection in aerial and satellite imagery
|
|
5
|
+
Author: Jeff Faudi, DL4EO
|
|
6
|
+
Maintainer: Jeff Faudi, DL4EO
|
|
7
|
+
License-Expression: Apache-2.0
|
|
8
|
+
Project-URL: Homepage, https://github.com/DL4EO/oriented-det
|
|
9
|
+
Project-URL: Documentation, https://github.com/DL4EO/oriented-det#readme
|
|
10
|
+
Project-URL: Repository, https://github.com/DL4EO/oriented-det
|
|
11
|
+
Project-URL: Issues, https://github.com/DL4EO/oriented-det/issues
|
|
12
|
+
Project-URL: Bug Tracker, https://github.com/DL4EO/oriented-det/issues
|
|
13
|
+
Project-URL: DL4EO, https://dl4eo.com
|
|
14
|
+
Keywords: computer-vision,object-detection,rotated-detection,oriented-detection,pytorch,aerial-imagery,satellite-imagery,remote-sensing,dota,geospatial
|
|
15
|
+
Classifier: Development Status :: 3 - Alpha
|
|
16
|
+
Classifier: Intended Audience :: Developers
|
|
17
|
+
Classifier: Intended Audience :: Science/Research
|
|
18
|
+
Classifier: Operating System :: OS Independent
|
|
19
|
+
Classifier: Programming Language :: Python :: 3
|
|
20
|
+
Classifier: Programming Language :: Python :: 3.10
|
|
21
|
+
Classifier: Programming Language :: Python :: 3.11
|
|
22
|
+
Classifier: Programming Language :: Python :: 3.12
|
|
23
|
+
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
|
|
24
|
+
Classifier: Topic :: Scientific/Engineering :: Image Recognition
|
|
25
|
+
Requires-Python: >=3.10
|
|
26
|
+
Description-Content-Type: text/markdown
|
|
27
|
+
License-File: LICENSE
|
|
28
|
+
Requires-Dist: torch>=2.3.0
|
|
29
|
+
Requires-Dist: torchvision>=0.18.0
|
|
30
|
+
Requires-Dist: numpy<2
|
|
31
|
+
Requires-Dist: Pillow>=9.0
|
|
32
|
+
Requires-Dist: albumentations>=1.3.0
|
|
33
|
+
Requires-Dist: tqdm>=4.65.0
|
|
34
|
+
Requires-Dist: tensorboard>=2.10
|
|
35
|
+
Requires-Dist: matplotlib>=3.5.0
|
|
36
|
+
Requires-Dist: huggingface_hub>=0.20.0
|
|
37
|
+
Requires-Dist: shapely>=2.0.0
|
|
38
|
+
Provides-Extra: export
|
|
39
|
+
Requires-Dist: onnx>=1.14.0; extra == "export"
|
|
40
|
+
Requires-Dist: onnxruntime>=1.16.0; extra == "export"
|
|
41
|
+
Requires-Dist: onnx2tf>=1.22.0; extra == "export"
|
|
42
|
+
Requires-Dist: tensorflow>=2.14.0; extra == "export"
|
|
43
|
+
Provides-Extra: dev
|
|
44
|
+
Requires-Dist: pytest>=8.0; extra == "dev"
|
|
45
|
+
Requires-Dist: pytest-cov>=4.0; extra == "dev"
|
|
46
|
+
Requires-Dist: black>=23.0; extra == "dev"
|
|
47
|
+
Requires-Dist: ruff>=0.1.0; extra == "dev"
|
|
48
|
+
Requires-Dist: build>=1.0; extra == "dev"
|
|
49
|
+
Requires-Dist: twine>=5.0; extra == "dev"
|
|
50
|
+
Provides-Extra: docs
|
|
51
|
+
Requires-Dist: mkdocs>=1.5.0; extra == "docs"
|
|
52
|
+
Requires-Dist: mkdocs-material>=9.0.0; extra == "docs"
|
|
53
|
+
Requires-Dist: mkdocstrings[python]>=0.23.0; extra == "docs"
|
|
54
|
+
Requires-Dist: pymdown-extensions>=10.0; extra == "docs"
|
|
55
|
+
Provides-Extra: all
|
|
56
|
+
Requires-Dist: oriented-det[dev]; extra == "all"
|
|
57
|
+
Dynamic: license-file
|
|
58
|
+
|
|
59
|
+
# OrientedDet
|
|
60
|
+
|
|
61
|
+
**OrientedDet** is a lightweight, modern PyTorch library for **rotated object detection** in aerial and satellite imagery. It focuses on clean geometry, reliable operators, simple datasets, and practical baseline models—without the complexity of large detection frameworks. OrientedDet is designed for researchers, practitioners, and geospatial developers who need accurate rotation-aware detectors with a minimal API.
|
|
62
|
+
|
|
63
|
+
## Features
|
|
64
|
+
|
|
65
|
+
- **Geometry**: Rotated bounding boxes (rbox: cx, cy, w, h, angle), quadrilateral boxes (qbox), polygon ↔ rbox ↔ hbox conversions, angle normalization (le90, 0–180°), flip/rotate/scale transforms, visualization helpers
|
|
66
|
+
- **IoU & NMS**: Rotated IoU and oriented NMS (CPU with optional GPU kernels when available); AABB pre-filtering; `obb_to_xyxy` / HBB conversion
|
|
67
|
+
- **Datasets**: DOTA polygon loader (pattern, split file, or separate folders), image tiling, label filtering, ignore masks, oriented mAP evaluation
|
|
68
|
+
- **Models**: **Oriented R-CNN** (horizontal RPN + MidpointOffset → oriented ROI), **Rotated Faster R-CNN** (oriented RPN + oriented ROI), **Rotated RetinaNet** (oriented anchors, focal loss); ResNet + FPN backbones; selective loading of external checkpoints where configs wire `checkpoint.load_from_checkpoint`
|
|
69
|
+
- **Training**: JSON configs + **`odet train`**, mixed precision (AMP), gradient accumulation, checkpointing, best-metric tracking, TensorBoard, optional curriculum learning and profiling
|
|
70
|
+
|
|
71
|
+
## Installation
|
|
72
|
+
|
|
73
|
+
- **Python** >= 3.9. Use [uv](https://docs.astral.sh/uv/) to install Python versions and manage the virtual environment.
|
|
74
|
+
- From the repo root (Linux with CUDA 12.1):
|
|
75
|
+
|
|
76
|
+
```bash
|
|
77
|
+
# Make sure you're in the project directory
|
|
78
|
+
cd ~/oriented-det
|
|
79
|
+
|
|
80
|
+
# Install uv if needed: https://docs.astral.sh/uv/getting-started/installation/
|
|
81
|
+
# Create a venv with Python 3.12 (uv downloads the interpreter if missing)
|
|
82
|
+
uv venv --python 3.12
|
|
83
|
+
source .venv/bin/activate
|
|
84
|
+
|
|
85
|
+
# Install PyTorch with CUDA 12.1 from PyTorch's index (2.3.0 or a higher version is fine)
|
|
86
|
+
uv pip install "torch>=2.3.0" "torchvision>=0.18.0" --index-url https://download.pytorch.org/whl/cu121
|
|
87
|
+
|
|
88
|
+
# Install dependencies and the project in editable mode
|
|
89
|
+
uv pip install -r requirements.txt
|
|
90
|
+
uv pip install -e .
|
|
91
|
+
```
|
|
92
|
+
|
|
93
|
+
- From PyPI (when published): `pip install oriented-det`
|
|
94
|
+
- For development and tests: `uv pip install -e ".[dev]"`
|
|
95
|
+
- For **macOS Apple Silicon** or **CPU-only**, see [Installation](docs/getting-started/installation.md).
|
|
96
|
+
- Verify: `pytest tests/test_geometry.py tests/test_iou.py tests/test_nms.py`
|
|
97
|
+
|
|
98
|
+
## Quick start
|
|
99
|
+
|
|
100
|
+
After [installation](#installation):
|
|
101
|
+
|
|
102
|
+
```bash
|
|
103
|
+
# Train on DOTA tiles (edit dataset paths in the config first)
|
|
104
|
+
odet train --config configs/rotated_faster_rcnn/dota_le90_3x.json
|
|
105
|
+
|
|
106
|
+
# Or use the Makefile wrapper (same default config)
|
|
107
|
+
make train
|
|
108
|
+
```
|
|
109
|
+
|
|
110
|
+
Default 3× recipe: `configs/rotated_faster_rcnn/dota_le90_3x.json` (see [configs/rotated_faster_rcnn/README.md](configs/rotated_faster_rcnn/README.md)). 1× baseline: `dota_le90_1x.json`.
|
|
111
|
+
|
|
112
|
+
Programmatic APIs and a longer walkthrough: [Getting Started](docs/getting-started/installation.md). Config fields: [Configuration](docs/user-guide/configuration.md).
|
|
113
|
+
|
|
114
|
+
### Paths in documentation and configs
|
|
115
|
+
|
|
116
|
+
Examples throughout this repo use placeholder paths such as **`/path/to/data`** and **`/path/to/oriented-det`**. You can point commands and JSON configs at your real locations (e.g. `dataset.data_root` in a training config), or keep those placeholders and map them with symbolic links:
|
|
117
|
+
|
|
118
|
+
```bash
|
|
119
|
+
# Create the parent directory (may need sudo for paths under /path/to)
|
|
120
|
+
sudo mkdir -p /path/to
|
|
121
|
+
|
|
122
|
+
# Point the placeholder at your DOTA dataset
|
|
123
|
+
ln -s /home/username/dota /path/to/data
|
|
124
|
+
|
|
125
|
+
# Point the placeholder at your clone of this repo
|
|
126
|
+
ln -s /home/username/oriented-det /path/to/oriented-det
|
|
127
|
+
```
|
|
128
|
+
|
|
129
|
+
After that, copy-pasted commands and unmodified configs that reference `/path/to/data` or `/path/to/oriented-det` resolve to your machine. Use relative symlink targets when you want the link to stay valid if the parent directory moves.
|
|
130
|
+
|
|
131
|
+
## Repository layout
|
|
132
|
+
|
|
133
|
+
| What | Where | You use it for |
|
|
134
|
+
|------|--------|----------------|
|
|
135
|
+
| **Library** | [`oriented_det/`](oriented_det/) | Geometry, models, datasets, training engine, ops — import in Python or extend in your own code |
|
|
136
|
+
| **CLI** | **`odet`** ([`oriented_det/cli/`](oriented_det/cli/)) | Train, tile data, run val inference, metrics, demos — **primary interface** after `uv pip install -e .` |
|
|
137
|
+
| **CLI implementations** | [`tools/`](tools/) | Python modules that implement `odet` subcommands (train, preds, tiling, …). Not a separate “old” API; contributors and debugging may call `python -m tools.train` directly |
|
|
138
|
+
| **Configs** | [`configs/`](configs/) | Experiment JSON (`_base_` inheritance, schema in `configs/config.schema.json`) |
|
|
139
|
+
| **Runs** | `runs/<model_type>/<timestamp>/` | Checkpoints, `config.json` snapshot, `train.log` (created at train time; not shipped in the repo) |
|
|
140
|
+
| **Docs** | [`docs/`](docs/) | MkDocs user guide and API reference |
|
|
141
|
+
| **Export** | [`export/`](export/) | Optional ONNX / TensorFlow export pipeline |
|
|
142
|
+
| **Examples** | [`demo/`](demo/), [`pretrained/`](pretrained/) | Demo images; registered checkpoints (large `.pth` files are usually gitignored) |
|
|
143
|
+
|
|
144
|
+
**`odet` vs `tools/`:** Installing the package registers the `odet` command. It loads modules under `tools/` (for example `tools.train`, `tools.save_predictions`). Shared inference and collate code lives in [`oriented_det/runtime/`](oriented_det/runtime/). You do not need two workflows — use **`odet`** (or **`make`**, which calls `odet`).
|
|
145
|
+
|
|
146
|
+
**Publishing a clean tree:** Ship the library, configs, docs, and tests. Omit local experiment output (`runs/`), datasets, and machine-specific paths in configs/Makefile.
|
|
147
|
+
|
|
148
|
+
## Documentation
|
|
149
|
+
|
|
150
|
+
Full documentation is in the **docs/** folder and can be built and served with MkDocs:
|
|
151
|
+
|
|
152
|
+
- **Build/serve**: `make docs` or `make docs-serve` (see [docs/README.md](docs/README.md)); or `uv pip install -e ".[docs]"` then `mkdocs serve`.
|
|
153
|
+
- **Guides**: [Getting Started](docs/getting-started/installation.md), [User Guide](docs/user-guide/geometry.md), [API Reference](docs/api/geometry.md), [Examples](docs/examples/visualization.md).
|
|
154
|
+
|
|
155
|
+
## Documentation by folder
|
|
156
|
+
|
|
157
|
+
| Folder | README | Description |
|
|
158
|
+
|--------|--------|-------------|
|
|
159
|
+
| [export/](export/) | [export/README.md](export/README.md) | Phase 1: PyTorch → ONNX → Keras detect bundle; `cd export && make export-tf` |
|
|
160
|
+
| [demo/](demo/) | [demo/README.md](demo/README.md) | Demo images; `odet image-demo` or `make demo` with the latest `runs/` checkpoint |
|
|
161
|
+
| [pretrained/](pretrained/) | [pretrained/README.md](pretrained/README.md) | Registered checkpoints for fine-tunes; large `.pth` files are usually gitignored |
|
|
162
|
+
| [oriented_det/cli/](oriented_det/cli/) | [oriented_det/cli/README.md](oriented_det/cli/README.md) | **`odet`** entry point and subcommand list |
|
|
163
|
+
| [tools/](tools/) | [tools/README.md](tools/README.md) | CLI script implementations (invoked by `odet`; see [Repository layout](#repository-layout)) |
|
|
164
|
+
| [configs/](configs/) | [configs/README.md](configs/README.md) | DOTA configs and **pretrain model zoo** (`_base_` inheritance) |
|
|
165
|
+
| [deploy/example/](deploy/example/) | [deploy/example/README.md](deploy/example/README.md) | Minimal DOTA deploy smoke image |
|
|
166
|
+
| [docs/](docs/) | [docs/README.md](docs/README.md) | MkDocs source; full user guide and API reference |
|
|
167
|
+
|
|
168
|
+
## Training and evaluation
|
|
169
|
+
|
|
170
|
+
Install once: `uv pip install -e .`. Then:
|
|
171
|
+
|
|
172
|
+
| Task | Command |
|
|
173
|
+
|------|---------|
|
|
174
|
+
| Train | `odet train --config configs/rotated_faster_rcnn/dota_le90_3x.json` or `make train` |
|
|
175
|
+
| Multi-GPU | `make train-multi-gpu` (`torchrun` + cuDNN on `LD_LIBRARY_PATH`) |
|
|
176
|
+
| Tile DOTA | `odet tile-dota /path/to/dota/train` |
|
|
177
|
+
| Val predictions | `odet preds --experiment-dir runs/rotated_faster_rcnn/<timestamp>` or `make preds` |
|
|
178
|
+
| Offline mAP | `make eval-val` or `make preds` then `make metrics` |
|
|
179
|
+
|
|
180
|
+
DOTA configs: per-model `dota_le90_1x.json` / `dota_le90_3x.json` under [configs/](configs/). Run `odet --help` for all subcommands. Makefile shortcuts and script-level options: [tools/README.md](tools/README.md). Config reference: [docs/user-guide/configuration.md](docs/user-guide/configuration.md), [configs/config.schema.json](configs/config.schema.json), [configs/README.md](configs/README.md).
|
|
181
|
+
|
|
182
|
+
## Pretrained weights and evaluation
|
|
183
|
+
|
|
184
|
+
- Place exported best checkpoints under **`pretrained/`** or use Hub slugs (`odet pretrained download oriented_rcnn_dota_le90_1x`). See [pretrained/README.md](pretrained/README.md) and [configs/README.md](configs/README.md#dota-pretrained-models-model-zoo).
|
|
185
|
+
- **Tiled validation:** after training, run `make preds` then `make metrics` (or `odet preds` / `odet preds --metrics-from-json`). Large images use sliding-window inference when they exceed the model canvas; thresholds and overlap come from **`production.*`** in the experiment config.
|
|
186
|
+
|
|
187
|
+
## Important notes
|
|
188
|
+
|
|
189
|
+
- **Angles**: Radians; use a single convention (e.g. le90 for DOTA). Helper: `normalize_le90` from `oriented_det.geometry`. For angle-delta normalization in custom code, see `oriented_det.models.oriented_rpn.normalize_angle_delta`.
|
|
190
|
+
- **NMS**: `torchvision.ops.nms_rotated` does not exist; the project uses a Python-based oriented NMS with AABB pre-filtering. GPU kernels are used when available.
|
|
191
|
+
- **ROI/memory**: Training samples 512 proposals per image; use `roi_chunk_size` and `roi_use_checkpoint` for memory tuning on large GPUs.
|
|
192
|
+
|
|
193
|
+
## Roadmap
|
|
194
|
+
|
|
195
|
+
- **v0.1** (current): Geometry, IoU/NMS, DOTA loader + tiler, three baseline detectors, config-based training, Hub pretrained weights
|
|
196
|
+
- **Next**: HRSC2016 dataset support, expanded tutorials, hosted docs
|
|
197
|
+
- **Future**: Additional oriented detectors (e.g. FAIR1M), optional fused CUDA kernels
|
|
198
|
+
|
|
199
|
+
## Contributing
|
|
200
|
+
|
|
201
|
+
Contributions are welcome. Run tests with `pytest`, format with `black` and `ruff`. See [docs/contributing.md](docs/contributing.md) for guidelines.
|
|
202
|
+
|
|
203
|
+
## Publishing to PyPI
|
|
204
|
+
|
|
205
|
+
Version **0.1.0** — tag releases as **`v0.1.0`** (git) matching `version` in `pyproject.toml`.
|
|
206
|
+
|
|
207
|
+
Configs: edit **`configs/`** at the repo root, then **`make sync-configs`** so **`oriented_det/configs/`** stays in sync (see **`oriented_det/configs/vendored_manifest.txt`**). CI runs **`make check-configs`**.
|
|
208
|
+
|
|
209
|
+
Install publishing tools (included in `.[dev]`):
|
|
210
|
+
|
|
211
|
+
```bash
|
|
212
|
+
uv pip install -e ".[dev]" # or: make publish-deps
|
|
213
|
+
```
|
|
214
|
+
|
|
215
|
+
### Step-by-step: first release (v0.1.0)
|
|
216
|
+
|
|
217
|
+
#### 1. One-time PyPI setup
|
|
218
|
+
|
|
219
|
+
1. Create accounts on [test.pypi.org](https://test.pypi.org) and [pypi.org](https://pypi.org) (can use the same email).
|
|
220
|
+
2. **Register the project name** `oriented-det` on PyPI (first upload creates it; TestPyPI is separate).
|
|
221
|
+
3. **Trusted Publishing (recommended)** — on each index, add a publisher for `DL4EO/oriented-det`, workflow `publish.yml`, environments `testpypi` / `pypi`. See [.github/workflows/README.md](.github/workflows/README.md).
|
|
222
|
+
4. **Or API tokens** — create `pypi-…` tokens and export locally (or add GitHub secrets `TESTPYPI_API_TOKEN`, `PYPI_API_TOKEN`):
|
|
223
|
+
|
|
224
|
+
```bash
|
|
225
|
+
export TWINE_USERNAME=__token__
|
|
226
|
+
export TWINE_PASSWORD=pypi-AgEI… # TestPyPI or PyPI token
|
|
227
|
+
```
|
|
228
|
+
|
|
229
|
+
#### 2. Pre-release checklist
|
|
230
|
+
|
|
231
|
+
```bash
|
|
232
|
+
make check-configs
|
|
233
|
+
pytest tests/ -q
|
|
234
|
+
```
|
|
235
|
+
|
|
236
|
+
Bump version in `pyproject.toml` and `docs/changelog.md` if needed (currently **0.1.0**).
|
|
237
|
+
|
|
238
|
+
#### 3. Build locally
|
|
239
|
+
|
|
240
|
+
```bash
|
|
241
|
+
make sync-configs # after changing configs/ or the manifest
|
|
242
|
+
make build # check-configs + sdist + wheel into dist/
|
|
243
|
+
make twine-check
|
|
244
|
+
```
|
|
245
|
+
|
|
246
|
+
Smoke-test the wheel:
|
|
247
|
+
|
|
248
|
+
```bash
|
|
249
|
+
python -m venv /tmp/odet-smoke && source /tmp/odet-smoke/bin/activate
|
|
250
|
+
pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
|
|
251
|
+
pip install dist/*.whl
|
|
252
|
+
odet --help
|
|
253
|
+
python -c "from oriented_det.geometry import RBox; print(RBox(0,0,1,1,0).area)"
|
|
254
|
+
```
|
|
255
|
+
|
|
256
|
+
#### 4. Upload to TestPyPI
|
|
257
|
+
|
|
258
|
+
```bash
|
|
259
|
+
make publish-testpypi
|
|
260
|
+
```
|
|
261
|
+
|
|
262
|
+
Install from TestPyPI (PyPI index still needed for dependencies like `torch`):
|
|
263
|
+
|
|
264
|
+
```bash
|
|
265
|
+
pip install --index-url https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple oriented-det==0.1.0
|
|
266
|
+
odet --help
|
|
267
|
+
```
|
|
268
|
+
|
|
269
|
+
Or trigger **Actions → Publish → Run workflow** (target: `testpypi`) after configuring Trusted Publishing or secrets.
|
|
270
|
+
|
|
271
|
+
#### 5. Tag and upload to production PyPI
|
|
272
|
+
|
|
273
|
+
When TestPyPI looks good:
|
|
274
|
+
|
|
275
|
+
```bash
|
|
276
|
+
git tag -a v0.1.0 -m "Release 0.1.0"
|
|
277
|
+
git push origin v0.1.0 # triggers publish.yml → PyPI (if CI is configured)
|
|
278
|
+
```
|
|
279
|
+
|
|
280
|
+
Local upload (fallback):
|
|
281
|
+
|
|
282
|
+
```bash
|
|
283
|
+
make publish-pypi
|
|
284
|
+
```
|
|
285
|
+
|
|
286
|
+
Create a [GitHub Release](https://github.com/DL4EO/oriented-det/releases) from tag `v0.1.0` with notes from `docs/changelog.md`.
|
|
287
|
+
|
|
288
|
+
#### 6. After publish
|
|
289
|
+
|
|
290
|
+
- Change install docs from “when published” to `pip install oriented-det`.
|
|
291
|
+
- Users still install **PyTorch** separately for their CUDA/CPU platform.
|
|
292
|
+
|
|
293
|
+
### Makefile targets
|
|
294
|
+
|
|
295
|
+
| Target | Action |
|
|
296
|
+
|--------|--------|
|
|
297
|
+
| `make publish-deps` | Install `build` and `twine` |
|
|
298
|
+
| `make build` | `check-configs` + `python -m build` |
|
|
299
|
+
| `make twine-check` | Validate `dist/*` metadata |
|
|
300
|
+
| `make publish-testpypi` | Build, check, upload to TestPyPI |
|
|
301
|
+
| `make publish-pypi` | Build, check, upload to PyPI |
|
|
302
|
+
|
|
303
|
+
Notes:
|
|
304
|
+
|
|
305
|
+
- **Trusted Publishing** avoids long-lived tokens; keep `make publish-*` for manual releases. Details: [.github/workflows/README.md](.github/workflows/README.md).
|
|
306
|
+
|
|
307
|
+
## License
|
|
308
|
+
|
|
309
|
+
**Apache-2.0** — Copyright © Jeff Faudi and [DL4EO](https://dl4eo.com). See [LICENSE](LICENSE) for details.
|
|
310
|
+
|
|
311
|
+
## Acknowledgements
|
|
312
|
+
|
|
313
|
+
Design and APIs are informed by MMRotate, MMDetection, Detectron2, and related work in oriented object detection; **pretrained checkpoints in `pretrained/` are OrientedDet exports from this codebase**, not MMRotate zoo bundles.
|
|
@@ -0,0 +1,115 @@
|
|
|
1
|
+
export/__init__.py,sha256=NZFHEjggrllpQwlJpPhj836_AJPCtSjDpVpetPFCJLk,189
|
|
2
|
+
export/ort_runtime.py,sha256=xU9QOd6An4LCaJFCG6rkU-E8lU2WsMTC-dqm4D7yt_4,2519
|
|
3
|
+
export/postprocess.py,sha256=nkRyciL3Hkjair02PyKrFK8_FQn3HLf0cOdu02qH0vs,5561
|
|
4
|
+
export/tf_serving_model.py,sha256=0svldtGyXYtuQiawy7ck1nTd8DDZqntlteGSR-5ZHhA,3345
|
|
5
|
+
export/val_dataset.py,sha256=Wr_0oQRTotse2UWkNYETzCNWiRfIBs62rqypuS-nk4E,5242
|
|
6
|
+
export/wrappers.py,sha256=B8t0NH6hhAx3cZMgWhU65AZybxZb34ma_5qCqJPjx18,6126
|
|
7
|
+
export/scripts/__init__.py,sha256=ml2MwROXcUgXIuj5qm6zE0Xxg4I2aeucHelJ2b51W5A,96
|
|
8
|
+
export/scripts/build_faster_rcnn_savedmodel.py,sha256=yag46p5VF6WRDhVR8NklvQJ6YXa5eUYVHfgfop3TSQQ,3479
|
|
9
|
+
export/scripts/export_onnx.py,sha256=85P2hrU1_eHuibX5iEijwoe2rKduXyKDSk3C6wvFYsM,7278
|
|
10
|
+
export/scripts/onnx_to_savedmodel.py,sha256=OeKddLMuhV39-lx7Cbgh0TZNFNr6JSya1eUTZ4wM9D8,1078
|
|
11
|
+
export/scripts/predict_savedmodel.py,sha256=f-d0VuzXofo1x8JxyQJVzhq1OoNq94pAn117BphQPus,3323
|
|
12
|
+
export/scripts/save_predictions_tf.py,sha256=2QmUo3ElCVD-NNNLLCXDNOuHm5n9RfGMw8T_13V-Oko,17589
|
|
13
|
+
export/scripts/to_tflite.py,sha256=CUiIWhb-FZQyWSwVhbMvQSOPuODL2jRkSNZfyEoWjn0,1002
|
|
14
|
+
export/tests/test_export_onnx_optional.py,sha256=Iuvwpm71ZBqjRC4TVILqYC5K8Q03YCc1DYAdv3rznD4,2458
|
|
15
|
+
export/tests/test_export_wrappers.py,sha256=Dv0UjnNLdvS63Ua-dcgZoEWjuZ3FtPVpHa_GXy-FyUI,3420
|
|
16
|
+
export/tests/test_faster_rcnn_export_parity.py,sha256=rFmBpSWjmhtLrkwELJMBgLdyANvXPLTqAQottbw8YVI,6602
|
|
17
|
+
export/tests/test_ort_runtime.py,sha256=LfANy-OvE3TatfPsnVev7Hl6gMu0mDmaaW2FTtcMD-w,1102
|
|
18
|
+
oriented_det/__init__.py,sha256=4U6f648If8d4QRRfSqI0OxBgZoxYnkwU5vMlMQyYu2s,2462
|
|
19
|
+
oriented_det/cli/__init__.py,sha256=5LGP4TQFKUqp4fcPDyaOuS4cvR2TdpAzzs1wQcog9P8,3497
|
|
20
|
+
oriented_det/cli/train.py,sha256=5Me3LZxU-Hl_xf1xRyYZwJx7fCmx0h-CTvFN9X35Fao,471
|
|
21
|
+
oriented_det/configs/config.schema.json,sha256=C5woB9ukzazc368528H8J_XNUqH2pKhqipl8UUIt6NA,41206
|
|
22
|
+
oriented_det/configs/_base_/augmentation.json,sha256=ELpx8TEDtZcBRPyFYnZ2xJczmWs3JtLUpbH7H3Llcls,355
|
|
23
|
+
oriented_det/configs/_base_/fp16.json,sha256=wNqdxfTeFIk50RVE30XBRhNiYSAHPnawoJ18rvg-VcQ,44
|
|
24
|
+
oriented_det/configs/_base_/preprocessing.json,sha256=NELtNcL_wTPAXwzMI9LMAAhG0XlIHjbolbGU7kmvhSA,161
|
|
25
|
+
oriented_det/configs/_base_/datasets/dota_le90.json,sha256=tMwaZiQtlEkgTKjFTYgXjd6NpaE4NUt5eOZ-c5CURkI,426
|
|
26
|
+
oriented_det/configs/_base_/models/oriented_rcnn_r50.json,sha256=2YvW8O3rJJ1If5PSILij2YmZbURH_dymCArU8MLIYUY,800
|
|
27
|
+
oriented_det/configs/_base_/models/rotated_faster_rcnn_r50.json,sha256=nrwyZYkIPbS5FlADr55rURKGFK0XcdgWWklDKJrJxKQ,1194
|
|
28
|
+
oriented_det/configs/_base_/models/rotated_retinanet_r50.json,sha256=fJg2PAcvgS7r9sxYpgvtvrqK9TxTKdvGjsdS58G3VJY,759
|
|
29
|
+
oriented_det/configs/_base_/schedules/1x.json,sha256=MNwVRKlK17ruJjxqwO1jUqL3YU95gMKeqJuIC2jyiME,754
|
|
30
|
+
oriented_det/configs/oriented_rcnn/dota_le90_1x.json,sha256=HJToX4Cn27_0VDPPd9OiuxZKo5oE-e7YSapVi13viUw,3308
|
|
31
|
+
oriented_det/configs/oriented_rcnn/dota_le90_3x.json,sha256=dYQpUfnhdmeO97sIlaOOwNhiiFSAWZJDz-kN4TcQYAI,188
|
|
32
|
+
oriented_det/configs/rotated_faster_rcnn/dota_le90_1x.json,sha256=fh6-2tbkylpPs_CXnzN6JBnu_XgBwR2ZqwsKNIyP4jw,3270
|
|
33
|
+
oriented_det/configs/rotated_faster_rcnn/dota_le90_3x.json,sha256=O5IE8qVBhDfwwB6h4gpmnEZf8wZHRaZdl9Bu3P5S3vg,182
|
|
34
|
+
oriented_det/configs/rotated_retinanet/dota_le90_1x.json,sha256=jjbNG3qJzikKkhA50GNa5-yhDs6BBm0_Y7acjnzx6Pg,2843
|
|
35
|
+
oriented_det/configs/rotated_retinanet/dota_le90_3x.json,sha256=xacNAkYrEePrFwZ7P43cKaCHcxVnDRwzG_q5C3kBv7E,720
|
|
36
|
+
oriented_det/data/__init__.py,sha256=6BGX-AQ2LqNPmmhlhwD_PSgZQRr28iOaCbEkXJUbR7c,2124
|
|
37
|
+
oriented_det/data/airbus_playground.py,sha256=qVINhAldCCHcxmx0U7zqjeZWx20vcOgfQUyLhv5ZsH4,22398
|
|
38
|
+
oriented_det/data/dota.py,sha256=2WF0zgJg-JBKKR9uz8LVyTP0kOQIzyeSfnx4SUHLmWo,29111
|
|
39
|
+
oriented_det/data/dota_classes.py,sha256=c45xxWp7AIX9DP-RI_NZ4iryo4z8a6nkN17E_-OdObY,643
|
|
40
|
+
oriented_det/data/evaluation.py,sha256=ALs35e94dLrZLc0w_Q2RiqDgEN33ELqoLzEEQ5FiIQk,28877
|
|
41
|
+
oriented_det/data/flips.py,sha256=YFJwyvCagWzE166vxJo-dYn6N1A9hxrRdXjmcF9Ty58,3670
|
|
42
|
+
oriented_det/data/preprocessing.py,sha256=Qkuh2e3NI2fxbGjL1sWSvzLRiu2yc2qYt_1S5kfMFb4,11153
|
|
43
|
+
oriented_det/data/tiling.py,sha256=Sp4UxMqCPybY8oFNKhnaMaEixW_GeJDcwrxOQdhJVRI,13687
|
|
44
|
+
oriented_det/data/transforms.py,sha256=oPdWrUqhC7iukpQrZGUG2b73sUGl_DtHXR0fex3MQtM,14015
|
|
45
|
+
oriented_det/geometry/__init__.py,sha256=rJs3GGihQfctCksBwW-HNaiCoByOnQl1P3iWIAXrWDA,274
|
|
46
|
+
oriented_det/geometry/poly.py,sha256=Y1bkTi6AM_qmI7Ezd1gPhWs8ADnnuLA-IlHF6gZSmSo,4174
|
|
47
|
+
oriented_det/geometry/qbox.py,sha256=evmpz7uFuUPQw3cYVmE5roCzSexRuCp1clk8q1kilSo,1926
|
|
48
|
+
oriented_det/geometry/rbox.py,sha256=EK8l29pW0-6NsTTMjJWFSpoW8pz_EKIWArc5WiB3IpY,8397
|
|
49
|
+
oriented_det/geometry/transforms.py,sha256=T1Ox5kSK3L7pPA4XGUQzPSh4I6hC5gtfykVtVG4mmRI,6976
|
|
50
|
+
oriented_det/models/__init__.py,sha256=FNSQjDYkHC-s5_rahXwA_TjlggtvyAVHnfR0k74gR-M,1011
|
|
51
|
+
oriented_det/models/bbox_coder.py,sha256=qPt3YF15gF1B1MnI_eUdAc1vhxOEcjVMAdAAKUfqtfo,12014
|
|
52
|
+
oriented_det/models/faster_rcnn_inference.py,sha256=WIzW5-xKlotNSOtODvUV0KoNS44Gbo-OAfweNhrb0jo,18422
|
|
53
|
+
oriented_det/models/horizontal_roi_coder.py,sha256=aF6C8vsDQCU6ap4bBRy_HS72in8mH3wbFLVgfP0JPII,5507
|
|
54
|
+
oriented_det/models/oriented_rcnn.py,sha256=7CmHlR_dAwXd8QkOr4rMn86hJQskLO8IlWP0pxlnrpk,67210
|
|
55
|
+
oriented_det/models/oriented_roi.py,sha256=j8vcWIav59b781NVmhTLfnfuv08fi1UcHhs66ujmquU,71016
|
|
56
|
+
oriented_det/models/oriented_rpn.py,sha256=ZxDhBbH3DAwyqPhfzxmeYWjtVvpUqkbLNNXF50tksLk,100493
|
|
57
|
+
oriented_det/models/rotated_retinanet.py,sha256=RyYNK_8fchfzL9pkdjf9RKdF_w8465aLVavQIpWSVgE,51376
|
|
58
|
+
oriented_det/models/utils.py,sha256=ZbV6Jtq2X2DiU6hPheKy7BrkIGK6Pqpo_RI4uhwwU3o,23165
|
|
59
|
+
oriented_det/models/backbones/__init__.py,sha256=pelzaP3J61M5jWk_9EIuLmQxMLB71eg4gV8KhRj26es,306
|
|
60
|
+
oriented_det/models/backbones/resnet_fpn.py,sha256=sn7eJJxpBLM4vXEi3Ss5GrqbdqDvWBcLpjjOBD1OaFI,3266
|
|
61
|
+
oriented_det/models/backbones/utils.py,sha256=dFt3U_AOSlHWLnDmB0phyWApjEZxBvh5mz70k5NmMnU,2620
|
|
62
|
+
oriented_det/ops/__init__.py,sha256=3iA7bKH_aHV3CbaLAiAa4VuZ5IZUfTJSn2VdWIVNbVc,1449
|
|
63
|
+
oriented_det/ops/gpu_ops.py,sha256=UpHcsXhhiuC0KUFZNParDAva7aPAO3H48hVSreT99Sw,41540
|
|
64
|
+
oriented_det/ops/iou.py,sha256=gHFhtb-4GFhcTsYCD2K3dJ08HzJM5LlZjCB3LAzckRE,5947
|
|
65
|
+
oriented_det/ops/kfiou.py,sha256=VxzQNL4xKKJE5l-SSFZUXZzlGYHzW8UhIj11-AyP9Zs,9674
|
|
66
|
+
oriented_det/ops/nms.py,sha256=gnGYAvR0JNeYNA3C1X9rZpyH6mASpR6kqE2uawiX1XM,7505
|
|
67
|
+
oriented_det/ops/probiou.py,sha256=s5QMtPio8eOUriTYzYSA5oeObOkxgzZ7mcnHh72oerw,5164
|
|
68
|
+
oriented_det/ops/rotated_ops.py,sha256=traolOPAUF-632fKP_u2R6TJvdST_b4m0_RLJ-RUNzg,4204
|
|
69
|
+
oriented_det/ops/utils.py,sha256=wfgZK6QQImq738NHu0fu83DhJXeai-gn3HfiZUcR3ao,8523
|
|
70
|
+
oriented_det/pretrained/__init__.py,sha256=eiOvmiVL5CJs7utmc8Uu_Y9VygNU3DMPAYduPWrCAlA,480
|
|
71
|
+
oriented_det/pretrained/hub.py,sha256=ho3kHGOIfPnBy8u7z0f-hhG_-Bulu8xXifEjiTSsCYM,8133
|
|
72
|
+
oriented_det/pretrained/manifest.json,sha256=aJ6hwajF9sqZhiEJLAaIW0Co8mgBrlFFkcMoSMOsk10,2360
|
|
73
|
+
oriented_det/runtime/__init__.py,sha256=nXje0BlmYYoaZB65tDMS9L5451Wq_H8drud6Yj6cx_Y,652
|
|
74
|
+
oriented_det/runtime/checkpoint.py,sha256=XagDEJK1qM3bshDYhU-tVy434Qm4otNbnxOW7Eyl0ik,15427
|
|
75
|
+
oriented_det/runtime/collate.py,sha256=2eCGTtoMBBBaIWfEMDenI17t8CAfKofTm4P3fnB1rbA,14679
|
|
76
|
+
oriented_det/runtime/inference.py,sha256=mMxidc9jLm0wIQVefidl3vxd6M5xSoEFNDfnbUlCgAQ,48905
|
|
77
|
+
oriented_det/train/__init__.py,sha256=qfj2It-nW8OiH0dZAdmliSe824EMlBEHsGNb9uzlUKY,2661
|
|
78
|
+
oriented_det/train/config.py,sha256=0CCs_RACt5RVn8u6C_c_CrbPXZ3SQpQMqNOxyKHmLlc,41844
|
|
79
|
+
oriented_det/train/engine.py,sha256=gV3OoRZR0ENtlnsMLo9uZOJw95a4XCaLMr6--08ohas,141572
|
|
80
|
+
oriented_det/train/grouped_ce.py,sha256=0lbJUTZcm57Njg1teU3Ww0pdGs_XTF3JnpFTAF19mYY,5022
|
|
81
|
+
oriented_det/train/piecewise_schedule.py,sha256=SAxMvo1Bv5UvFugJCThQ2rEqDxO4OAAAFFa6SeENuRY,1038
|
|
82
|
+
oriented_det/train/profiler.py,sha256=7Em9VQSqy534T3zJ-W7QJVo7T4QUxTdOZg_r_-ryo-8,9187
|
|
83
|
+
oriented_det/train/utils.py,sha256=VJhXEsDVIFhia2k_e6_S2RoN3b17CBqjTqlWQpoA89U,41030
|
|
84
|
+
oriented_det/utils/__init__.py,sha256=_dVKwmq7OUcmE9ChyWO8M9rC6hpAGdv3WC0OTwawSeU,693
|
|
85
|
+
oriented_det/utils/config.py,sha256=J8FEB01Vr1N8gGOG-ZKG0PWbuYJ2Hc2NiGfS50z512c,12367
|
|
86
|
+
oriented_det/utils/device.py,sha256=1d_JTdGb2TBySDYiKTvShhwjO-U8TzpEZio0HucVnRA,1285
|
|
87
|
+
oriented_det/utils/logging.py,sha256=Bn2AKLzNyXKagFf6TwU79sUI7kMstbUZcej5jK0eomc,5037
|
|
88
|
+
oriented_det/utils/progress.py,sha256=QUck27RMxzH_BBLjTmnuy2nlcGwD4D7SJrHJpNUc_Io,2144
|
|
89
|
+
oriented_det/utils/viz.py,sha256=yqWXxqBIZuDJaRdVLA4BWwkxvW3DZKQH73AUGl2tyoE,6008
|
|
90
|
+
oriented_det-0.1.0.dist-info/licenses/LICENSE,sha256=zWrwvHYHK37sq-iBd74vqxl6JwGxaaYrgFuV-3yuX74,11371
|
|
91
|
+
tools/__init__.py,sha256=g0EGO5OnTpFAUtBoSs9mHGzgHte_jxYygBQhVJNYKwk,79
|
|
92
|
+
tools/app.py,sha256=BLjNL4F2pFCSQENNMRWPrJPnAyD2E_lb0wizknXWMUg,54023
|
|
93
|
+
tools/dataset_stats.py,sha256=nn_HFr1KaZ1z9LjZdEeS30LXxn09tQH4QoIvESxi4Og,15773
|
|
94
|
+
tools/dota_labels_to_comma.py,sha256=b8Ku3_n2AbYAEdzcRNIu9G7GSqsyZlyig-nAyyidJRs,4663
|
|
95
|
+
tools/free_gpu.py,sha256=E6tM9K7WNdrQkzTcPZSNS3OIP1y4mGQlr9gQtuoDiMk,4318
|
|
96
|
+
tools/generate_airbus_playground_csv.py,sha256=4KoSnt3mqVdVkgVgnnJ_4dvFfDeuvntHnw7soRENong,5393
|
|
97
|
+
tools/image_demo.py,sha256=KqAloSSr84fwPg5a7SasMK8ttONuYaVeih_PASsh9UA,7238
|
|
98
|
+
tools/lr_finder.py,sha256=yNYgHqFnePJE30t1fwjq0qARKzDhBBZUM6Z6PamGfJs,29259
|
|
99
|
+
tools/measure_sampled_riou_error.py,sha256=VLV4hBsAPsvsG0bZdw_JL2qaDVS0XDd3ILIfdXn8HfQ,15334
|
|
100
|
+
tools/playground_to_dota.py,sha256=8x1tq1PgHe1ZD35JGAuXQNKl3tJqNxX7ac3REUKwJv8,11109
|
|
101
|
+
tools/pretrained_download.py,sha256=6V9xG5vAQw9nJvReyKmkWSriPViOwWQ2ie8SB9Q8jFw,1447
|
|
102
|
+
tools/preview_augmentation.py,sha256=97t41BwGP1H6K4VS-rjo__OLvk1XIkrjOa0OHyrEBFg,17361
|
|
103
|
+
tools/publish_checkpoint.py,sha256=TJvyc8EMnWaIBYMFE0JlEZyFHgnuW_BiNQFhH0K-Kw4,3025
|
|
104
|
+
tools/save_predictions.py,sha256=3SQuLxLoNVyHLYeqtVxdIb8GbAZYfYLk4Vghfp2WVLs,105316
|
|
105
|
+
tools/sync_vendored_configs.py,sha256=W_zwvKuLqOOsO1ZleL_Y9BQk5atTdKt5-OMY2vhJguY,2906
|
|
106
|
+
tools/tile_dota.py,sha256=kIzypIa36iUID8k_nUTYgg0KRvDAfrm-SpnatM3MQQ8,16737
|
|
107
|
+
tools/train.py,sha256=hhjLM1slqn_tOwxU4OLGiOgrU5YM1ympzHbDTMYD5H0,109037
|
|
108
|
+
tools/train_example.py,sha256=TT4kIy84D291erJxSh-Vi_EUidP06w27B9-U2yLLcYo,8841
|
|
109
|
+
tools/train_multi_gpu.py,sha256=rWhqzmz4r_6iduoIe4nGFMWTYAewOEcQM_bGlnhyW-I,15029
|
|
110
|
+
tools/visualize_boxes.py,sha256=888tmG2lst1Ijm-sFf2guw-QOFUylKf1iZ1u_0ECq54,6060
|
|
111
|
+
oriented_det-0.1.0.dist-info/METADATA,sha256=enSXiqyZC8VbzIeR9cFeBCrZx8pSBKKl9vbU_cr4hBs,16194
|
|
112
|
+
oriented_det-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
|
|
113
|
+
oriented_det-0.1.0.dist-info/entry_points.txt,sha256=KL0zwJ__y-IOna-3pA-bl5KZ6SRtrml78gNsDG762kQ,47
|
|
114
|
+
oriented_det-0.1.0.dist-info/top_level.txt,sha256=Ps78oOIJm0VxN9OeyRHCZ4wrb1Z3rSW392ybT6Up0kQ,26
|
|
115
|
+
oriented_det-0.1.0.dist-info/RECORD,,
|