clarify scale

mwalmsley · Mar 21, 2024 · 3347aa0 · 3347aa0
1 parent 77414a1
commit 3347aa0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/galaxy_datasets/transforms.py b/galaxy_datasets/transforms.py
@@ -8,7 +8,7 @@ def default_transforms(
     crop_ratio_bounds=(0.9, 1.1),
     resize_after_crop=224, 
     pytorch_greyscale=False,
-    to_float=False  # later zoobot checkpoints expect 0-1 float, not 0-255 float
+    to_float=True  # set to True when loading images directly, False via webdatasets (which normalizes to 0-1 on decode)
     ) -> A.Compose:
 
     transforms_to_apply = base_transforms(pytorch_greyscale)

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="galaxy-datasets",
-    version="0.0.16",
+    version="0.0.17",
     author="Mike Walmsley",
     author_email="[email protected]",
     description="Galaxy Zoo datasets for PyTorch/TensorFlow",