qingke1 commited on
Commit
8794fc2
·
1 Parent(s): 6b2e256

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