Spaces:
Sleeping
Sleeping
Update nnunetv2/inference/predict_from_raw_data.py
Browse files
nnunetv2/inference/predict_from_raw_data.py
CHANGED
|
@@ -37,9 +37,9 @@ import pickle
|
|
| 37 |
|
| 38 |
class nnUNetPredictor(object):
|
| 39 |
def __init__(self,
|
| 40 |
-
tile_step_size: float = 0.
|
| 41 |
use_gaussian: bool = True,
|
| 42 |
-
use_mirroring: bool =
|
| 43 |
perform_everything_on_device: bool = True,
|
| 44 |
device: torch.device = torch.device('cuda'),
|
| 45 |
verbose: bool = False,
|
|
|
|
| 37 |
|
| 38 |
class nnUNetPredictor(object):
|
| 39 |
def __init__(self,
|
| 40 |
+
tile_step_size: float = 0.75,
|
| 41 |
use_gaussian: bool = True,
|
| 42 |
+
use_mirroring: bool = False,
|
| 43 |
perform_everything_on_device: bool = True,
|
| 44 |
device: torch.device = torch.device('cuda'),
|
| 45 |
verbose: bool = False,
|