Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

BeninSmallHolderCashews: return geospatial metadata in sample #377

Merged
merged 13 commits into from
Feb 24, 2022
Merged

BeninSmallHolderCashews: return geospatial metadata in sample #377

Changes from 2 commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
299230a
exposing geoinformation
recursix Jan 29, 2022
2ba55e0
restoring white spaces
recursix Jan 29, 2022
8a00027
Returns tile_transform and crs
recursix Feb 1, 2022
eb81ce4
Merge branch 'geo-information' of https://github.com/recursix/torchge…
recursix Feb 1, 2022
2c830e5
adding types
recursix Feb 1, 2022
981f2f8
adjust line length
recursix Feb 1, 2022
092cf5f
fixing formatting
recursix Feb 2, 2022
5abbbc9
remove auto-format from another auto-pep8 tool
recursix Feb 3, 2022
7483fb8
adjust order of import
recursix Feb 3, 2022
21d3239
radjust autopep8
recursix Feb 3, 2022
4b159e7
passes pre-commit hook
recursix Feb 9, 2022
0485b3f
Add Oxford comma
adamjstewart Feb 23, 2022
11d5e6c
tile_transform -> transform
adamjstewart Feb 24, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion torchgeo/datasets/benin_cashews.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -233,7 +233,7 @@ def __getitem__(self, index: int) -> Dict[str, Tensor]:
index: index to return

Returns:
image, mask, and metadata at that index
image, mask, geotransform, crs and metadata at that index
adamjstewart marked this conversation as resolved.
Show resolved Hide resolved
"""
y, x = self.chips_metadata[index]

Expand All @@ -248,6 +248,8 @@ def __getitem__(self, index: int) -> Dict[str, Tensor]:
"mask": labels,
"x": torch.tensor(x), # type: ignore[attr-defined]
"y": torch.tensor(y), # type: ignore[attr-defined]
"geotransform": self.tile_transform,
adamjstewart marked this conversation as resolved.
Show resolved Hide resolved
"crs": self.crs
}

if self.transforms is not None:
Expand Down Expand Up @@ -343,6 +345,7 @@ def _load_single_scene(self, date: str, bands: Tuple[str, ...]) -> Tensor:
)
with rasterio.open(filepath) as src:
self.tile_transform = src.transform
self.crs = src.crs
calebrob6 marked this conversation as resolved.
Show resolved Hide resolved
array = src.read().astype(np.float32)
img[band_index] = torch.from_numpy(array) # type: ignore[attr-defined]

Expand Down