diff --git a/setup.py b/setup.py index 55b9e2b..72296db 100644 --- a/setup.py +++ b/setup.py @@ -11,6 +11,8 @@ install_requires=[ "torch>=1.10.0", "torchvision", + "torchgeo", # Added New + "open_clip_torch", # Added New "pandas", "geopandas", "numpy", diff --git a/src/embedder/models/SSL4EO_S2L1C.py b/src/embedder/models/SSL4EO_S2L1C.py index fad28dd..faea563 100644 --- a/src/embedder/models/SSL4EO_S2L1C.py +++ b/src/embedder/models/SSL4EO_S2L1C.py @@ -80,7 +80,7 @@ def preprocess(self, input): Returns: torch.Tensor: Preprocessed input, scaled by a factor of 10,000. """ - return input / 1e4 + return input / 1e4 def forward(self, input): """