|
28 | 28 | import sys |
29 | 29 | import logging |
30 | 30 | import argparse |
31 | | -import multiprocessing as mproc |
32 | 31 | from functools import partial |
33 | 32 |
|
34 | 33 | import tqdm |
|
73 | 72 | NAME_YAML_PARAMS = 'configuration.yaml' |
74 | 73 | NAME_DUMP_TRAIN_DATA = 'dump_training_data.npz' |
75 | 74 |
|
76 | | -NB_THREADS = max(1, int(mproc.cpu_count() * 0.9)) |
| 75 | +NB_WORKERS = tl_expt.nb_workers(0.9) |
77 | 76 | # position is label in loaded segm and nb are out labels |
78 | 77 | LUT_ANNOT_CENTER_RELABEL = [0, 0, -1, 1] |
79 | 78 | CROSS_VAL_LEAVE_OUT_SEARCH = 0.2 |
@@ -149,7 +148,7 @@ def arg_parse_params(params): |
149 | 148 | help='name of the experiment', default='ovary') |
150 | 149 | parser.add_argument('-cfg', '--path_config', type=str, required=False, |
151 | 150 | help='path to the configuration', default=None) |
152 | | - parser.add_argument('--nb_workers', type=int, required=False, default=NB_THREADS, |
| 151 | + parser.add_argument('--nb_workers', type=int, required=False, default=NB_WORKERS, |
153 | 152 | help='number of processes in parallel') |
154 | 153 | params.update(vars(parser.parse_args())) |
155 | 154 | paths = {} |
@@ -445,7 +444,7 @@ def wrapper_draw_export_slic_centers(args): |
445 | 444 | return export_show_image_points_labels(*args) |
446 | 445 |
|
447 | 446 |
|
448 | | -def dataset_load_images_segms_compute_features(params, df_paths, nb_workers=NB_THREADS): |
| 447 | +def dataset_load_images_segms_compute_features(params, df_paths, nb_workers=NB_WORKERS): |
449 | 448 | """ create whole dataset composed from loading input data, computing features |
450 | 449 | and label points by label whether its positive or negative center candidate |
451 | 450 |
|
@@ -502,7 +501,7 @@ def dataset_load_images_segms_compute_features(params, df_paths, nb_workers=NB_T |
502 | 501 |
|
503 | 502 |
|
504 | 503 | def export_dataset_visual(path_output, dict_imgs, dict_segms, dict_slics, |
505 | | - dict_points, dict_labels, nb_workers=NB_THREADS): |
| 504 | + dict_points, dict_labels, nb_workers=NB_WORKERS): |
506 | 505 | """ visualise complete training dataset by marking labeld points |
507 | 506 | over image and input segmentation |
508 | 507 |
|
|
0 commit comments