Spaces:
Running
Running
update monuseg
Browse files
cell_segmentation/inference/inference_cellvit_experiment_monuseg.py
CHANGED
@@ -94,19 +94,6 @@ class MoNuSegInference:
|
|
94 |
self.__load_model()
|
95 |
# self.__load_inference_transforms()
|
96 |
self.__setup_amp()
|
97 |
-
self.inference_dataset = MoNuSegDataset(
|
98 |
-
dataset_path=dataset_path,
|
99 |
-
transforms=self.inference_transforms,
|
100 |
-
patching=patching,
|
101 |
-
overlap=overlap,
|
102 |
-
)
|
103 |
-
self.inference_dataloader = DataLoader(
|
104 |
-
self.inference_dataset,
|
105 |
-
batch_size=1,
|
106 |
-
num_workers=8,
|
107 |
-
pin_memory=False,
|
108 |
-
shuffle=False,
|
109 |
-
)
|
110 |
|
111 |
def __instantiate_logger(self) -> None:
|
112 |
"""Instantiate logger
|
|
|
94 |
self.__load_model()
|
95 |
# self.__load_inference_transforms()
|
96 |
self.__setup_amp()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
97 |
|
98 |
def __instantiate_logger(self) -> None:
|
99 |
"""Instantiate logger
|