Commit
·
7a36946
1
Parent(s):
4dee018
fix bug
Browse files
app.py
CHANGED
|
@@ -129,7 +129,7 @@ def load_example(file_paths: List[str], mean: List[float], std: List[float]):
|
|
| 129 |
|
| 130 |
for file in file_paths:
|
| 131 |
img, meta = read_geotiff(file)
|
| 132 |
-
img = img[:6] if img[:6].mean() <= 2 else img[:6]
|
| 133 |
|
| 134 |
# Rescaling (don't normalize on nodata)
|
| 135 |
img = np.moveaxis(img, 0, -1) # channels last for rescaling
|
|
|
|
| 129 |
|
| 130 |
for file in file_paths:
|
| 131 |
img, meta = read_geotiff(file)
|
| 132 |
+
img = img[:6]*10000 if img[:6].mean() <= 2 else img[:6]
|
| 133 |
|
| 134 |
# Rescaling (don't normalize on nodata)
|
| 135 |
img = np.moveaxis(img, 0, -1) # channels last for rescaling
|