Skip to content

Commit

Permalink
Add version strings
Browse files Browse the repository at this point in the history
  • Loading branch information
ashnair1 committed Jan 10, 2024
1 parent 14edc17 commit ac5f80e
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions torchgeo/datamodules/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,10 @@ class MisconfigurationException(Exception):


class AugPipe(Module):
"""Pipeline for applying augmentations sequentially on select data keys."""
"""Pipeline for applying augmentations sequentially on select data keys.
.. versionadded:: 0.6
"""

def __init__(
self, augs: Callable[[dict[str, Any]], dict[str, Any]], batch_size: int
Expand Down Expand Up @@ -74,13 +77,15 @@ def forward(self, batch: dict[str, Tensor]) -> dict[str, Tensor]:


def collate_fn_detection(batch: list[dict[str, Tensor]]) -> dict[str, Any]:
"""Custom collate fn for object detection & instance segmentation.
"""Custom collate fn for object detection and instance segmentation.
Args:
batch: list of sample dicts return by dataset
Returns:
batch dict output
.. versionadded:: 0.6
"""
output: dict[str, Any] = {}
output["image"] = [sample["image"] for sample in batch]
Expand Down

0 comments on commit ac5f80e

Please sign in to comment.