|
346 | 346 | " ),\n", |
347 | 347 | " AsDiscreted(keys=pred_key, argmax=True),\n", |
348 | 348 | " ]\n", |
349 | | - " \n", |
350 | | - " transform_list.append(\n", |
351 | | - " SaveImaged(\n", |
352 | | - " keys=pred_key,\n", |
353 | | - " output_dir=output_dir,\n", |
354 | | - " output_ext=output_ext,\n", |
355 | | - " output_dtype=output_dtype,\n", |
356 | | - " output_postfix=output_postfix,\n", |
357 | | - " separate_folder=separate_folder,\n", |
| 349 | + "\n", |
| 350 | + " (\n", |
| 351 | + " transform_list.append(\n", |
| 352 | + " SaveImaged(\n", |
| 353 | + " keys=pred_key,\n", |
| 354 | + " output_dir=output_dir,\n", |
| 355 | + " output_ext=output_ext,\n", |
| 356 | + " output_dtype=output_dtype,\n", |
| 357 | + " output_postfix=output_postfix,\n", |
| 358 | + " separate_folder=separate_folder,\n", |
| 359 | + " )\n", |
358 | 360 | " )\n", |
359 | | - " )if save_output else transform_list\n", |
| 361 | + " if save_output\n", |
| 362 | + " else transform_list\n", |
| 363 | + " )\n", |
360 | 364 | " postprocessing_transforms = Compose(transform_list)\n", |
361 | 365 | " return postprocessing_transforms\n", |
362 | 366 | "\n", |
|
421 | 425 | " for key, value in kwargs.items():\n", |
422 | 426 | " if key not in self.POSTPROCESSING_EXTRA_ARGS:\n", |
423 | 427 | " logging.warning(f\"Cannot set parameter {key} for postprocessing.\")\n", |
424 | | - " if (\n", |
425 | | - " key in self._postprocess_params\n", |
426 | | - " and value != self._postprocess_params[key]\n", |
427 | | - " ) or (key not in self._postprocess_params):\n", |
| 428 | + " if (key in self._postprocess_params and value != self._postprocess_params[key]) or (\n", |
| 429 | + " key not in self._postprocess_params\n", |
| 430 | + " ):\n", |
428 | 431 | " self._postprocess_params.update(kwargs)\n", |
429 | | - " self.postprocessing_transforms = self._init_postprocessing_transforms(\n", |
430 | | - " **self._postprocess_params\n", |
431 | | - " )\n", |
| 432 | + " self.postprocessing_transforms = self._init_postprocessing_transforms(**self._postprocess_params)\n", |
432 | 433 | " outputs = decollate_batch(outputs)\n", |
433 | 434 | " outputs = self.postprocessing_transforms(outputs)\n", |
434 | 435 | " return outputs" |
|
0 commit comments