ChrisXzZ commited on
Commit
2c95083
·
verified ·
1 Parent(s): 48d78ac

Update masks_BDMAP.py

Browse files
Files changed (1) hide show
  1. masks_BDMAP.py +2 -2
masks_BDMAP.py CHANGED
@@ -61,10 +61,10 @@ def process_all_cases(source_dir, output_dir):
61
  cases = [f for f in os.listdir(source_dir) if f.endswith(".nii.gz")]
62
 
63
  print(f">> finding {len(cases)} cases, starting...")
64
- print(">> {} CPU cores are secured.".format(max(1, int(cpu_count() * 0.9))))
65
 
66
  results = []
67
- with ProcessPoolExecutor(max_workers=max(1, int(cpu_count() * 0.9))) as executor:
68
  futures = {
69
  executor.submit(process_case, case, source_dir, output_dir): case
70
  for case in cases
 
61
  cases = [f for f in os.listdir(source_dir) if f.endswith(".nii.gz")]
62
 
63
  print(f">> finding {len(cases)} cases, starting...")
64
+ print(">> {} CPU cores are secured.".format(max(1, int(cpu_count() * 0.8))))
65
 
66
  results = []
67
+ with ProcessPoolExecutor(max_workers=max(1, int(cpu_count() * 0.8))) as executor:
68
  futures = {
69
  executor.submit(process_case, case, source_dir, output_dir): case
70
  for case in cases