Upload 2 files

satlas/run.py

@@ -21,10 +21,10 @@ results = run_satlas(
 
 # Display the results
 fig, ax = plt.subplots(1, 3, figsize=(10, 5))
-ax[0].imshow(results["lr"].transpose(1, 2, 0)/10000)
+ax[0].imshow(results["lr"].transpose(1, 2, 0)/3000)
 ax[0].set_title("LR")
 ax[0].axis("off")
-ax[1].imshow(results["sr"].transpose(1, 2, 0)/10000)
+ax[1].imshow(results["sr"].transpose(1, 2, 0)/3000)
 ax[1].set_title("SR")
 ax[1].axis("off")
 ax[2].imshow(results["hr"].transpose(1, 2, 0) / 3000)

satlas/utils.py

@@ -40,7 +40,7 @@ def run_satlas(
     overlap: int = 0,
     device: Union[str, torch.device] = "cuda"
 ) -> torch.Tensor:
-    # Load the LR image
+    # Load the LR image 
     lr = torch.from_numpy(lr[[3, 2, 1]]/3558).float().to(device).clamp(0, 1)
 
     # Select the raster with the lowest resolution
@@ -85,9 +85,10 @@ def run_satlas(
     
     # Save the result
     results = {
-        "lr": (lr.cpu().numpy() * 10000).astype(np.uint16),
-        "sr": (sr.cpu().numpy() * 10000).astype(np.uint16),
+        "lr": (lr.cpu().numpy() * 3558).astype(np.uint16),
+        "sr": (sr.cpu().numpy() * 3558).astype(np.uint16),
         "hr": hr[0:3]
     }
+    
 
     return results