Update masks_BDMAP.py
Browse files- 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.
|
| 65 |
|
| 66 |
results = []
|
| 67 |
-
with ProcessPoolExecutor(max_workers=max(1, int(cpu_count() * 0.
|
| 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
|