Commit
•
5323997
1
Parent(s):
282b197
Fix img_size
Browse files
app.py
CHANGED
|
@@ -264,7 +264,7 @@ def predict_on_images(data_files: list, mask_ratio: float, yaml_file_path: str,
|
|
| 264 |
# data related
|
| 265 |
train_params = params["train_params"]
|
| 266 |
num_frames = model_params['num_frames']
|
| 267 |
-
img_size =
|
| 268 |
bands = train_params['bands']
|
| 269 |
mean = train_params['data_mean']
|
| 270 |
std = train_params['data_std']
|
|
|
|
| 264 |
# data related
|
| 265 |
train_params = params["train_params"]
|
| 266 |
num_frames = model_params['num_frames']
|
| 267 |
+
img_size = model_params['img_size']
|
| 268 |
bands = train_params['bands']
|
| 269 |
mean = train_params['data_mean']
|
| 270 |
std = train_params['data_std']
|