Skip to content

Commit

Permalink
removed integer images from documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
David Meaux committed Jan 17, 2024
1 parent 1fe44cb commit 635b383
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 6 deletions.
2 changes: 1 addition & 1 deletion docs/tutorials/custom_raster_dataset.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -333,7 +333,7 @@
"\n",
"### `dtype`\n",
"\n",
"Defaults to float32 for `is_image == True` and long for `is_image == False`. This is what is usually wanted for 99% of datasets but can be overridden for integer images (like some Digital Elevation Models) or pixel-wise regression masks (where the target should be float32). Uint16 and uint32 are automatically cast to int32 and int64, respectively, because numpy supports the former but torch does not.\n",
"Defaults to float32 for `is_image == True` and long for `is_image == False`. This is what is usually wanted for 99% of datasets but can be overridden for pixel-wise regression masks (where the target should be float32). Uint16 and uint32 are automatically cast to int32 and int64, respectively, because numpy supports the former but torch does not.\n",
"\n",
"### `separate_files`\n",
"\n",
Expand Down
9 changes: 4 additions & 5 deletions torchgeo/datasets/geo.py
Original file line number Diff line number Diff line change
Expand Up @@ -370,11 +370,10 @@ def dtype(self) -> torch.dtype:
"""The dtype of the dataset (overrides the dtype of the data file via a cast).
Defaults to float32 for is_image = True and long for is_image = False. This is
what we usually want for 99% of datasets but can be overridden for integer
images (like some imagery or Digital Elevation Models) or pixel-wise regression
masks (where it should be float32). Uint16 and uint32 are automatically cast to
int32 and int64, respectively, because numpy supports the former but torch does
not.
what we usually want for 99% of datasets but can be overridden for pixel-wise
regression masks (where it should be float32). Uint16 and uint32 are
automatically cast to int32 and int64, respectively, because numpy supports
the former but torch does not.
Returns:
the dtype of the dataset
Expand Down

0 comments on commit 635b383

Please sign in to comment.