Commit
·
40a835d
1
Parent(s):
921b18d
Fix date from file
Browse files- inference.py +3 -3
inference.py
CHANGED
|
@@ -140,7 +140,7 @@ def load_example(
|
|
| 140 |
location_coords.append(coords)
|
| 141 |
|
| 142 |
try:
|
| 143 |
-
match = re.search(r'(\d{7}T\d{6})', file)
|
| 144 |
if match:
|
| 145 |
year = int(match.group(1)[:4])
|
| 146 |
julian_day = match.group(1).split('T')[0][4:]
|
|
@@ -485,13 +485,13 @@ if __name__ == "__main__":
|
|
| 485 |
"--config_path",
|
| 486 |
"-c",
|
| 487 |
type=str,
|
| 488 |
-
default="
|
| 489 |
help="Path to yaml file containing model training parameters.",
|
| 490 |
)
|
| 491 |
parser.add_argument(
|
| 492 |
"--checkpoint",
|
| 493 |
type=str,
|
| 494 |
-
default="
|
| 495 |
help="Path to a checkpoint file to load from.",
|
| 496 |
)
|
| 497 |
parser.add_argument(
|
|
|
|
| 140 |
location_coords.append(coords)
|
| 141 |
|
| 142 |
try:
|
| 143 |
+
match = re.search(r'(\d{7,8}T\d{6})', file)
|
| 144 |
if match:
|
| 145 |
year = int(match.group(1)[:4])
|
| 146 |
julian_day = match.group(1).split('T')[0][4:]
|
|
|
|
| 485 |
"--config_path",
|
| 486 |
"-c",
|
| 487 |
type=str,
|
| 488 |
+
default="Prithvi_EO_V2_300M_TL_config.yaml",
|
| 489 |
help="Path to yaml file containing model training parameters.",
|
| 490 |
)
|
| 491 |
parser.add_argument(
|
| 492 |
"--checkpoint",
|
| 493 |
type=str,
|
| 494 |
+
default="Prithvi_EO_V2_300M_TL.pt",
|
| 495 |
help="Path to a checkpoint file to load from.",
|
| 496 |
)
|
| 497 |
parser.add_argument(
|