Commit 2bd6ea8b authored by tim.scherr's avatar tim.scherr
Browse files

cell type bugfix (foi correction)

parent 8e2efaac
...@@ -4,6 +4,7 @@ import random ...@@ -4,6 +4,7 @@ import random
import torch import torch
import warnings import warnings
from copy import deepcopy
from pathlib import Path from pathlib import Path
from segmentation.inference.inference import inference_2d_ctc, inference_3d_ctc from segmentation.inference.inference import inference_2d_ctc, inference_3d_ctc
...@@ -78,13 +79,16 @@ def main(): ...@@ -78,13 +79,16 @@ def main():
print(' Segmentation results already exist (delete for new calculation).') print(' Segmentation results already exist (delete for new calculation).')
continue continue
inference_args = deepcopy(args)
inference_args.cell_type = ct
if '2D' in ct: if '2D' in ct:
inference_2d_ctc(model=model, inference_2d_ctc(model=model,
data_path=path_data / ct / subset, data_path=path_data / ct / subset,
result_path=path_seg_results, result_path=path_seg_results,
device=device, device=device,
batchsize=args.batch_size, batchsize=args.batch_size,
args=args, args=inference_args,
num_gpus=num_gpus) num_gpus=num_gpus)
else: else:
inference_3d_ctc(model=model, inference_3d_ctc(model=model,
...@@ -92,7 +96,7 @@ def main(): ...@@ -92,7 +96,7 @@ def main():
result_path=path_seg_results, result_path=path_seg_results,
device=device, device=device,
batchsize=args.batch_size, batchsize=args.batch_size,
args=args, args=inference_args,
num_gpus=num_gpus) num_gpus=num_gpus)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment