(
self,
dataset,
class_num,
image_mean,
image_std,
network,
multi_scales,
is_flip,
devices,
verbose=False,
save_path=None,
show_image=False,
)
| 18 | |
| 19 | class Evaluator(object): |
| 20 | def __init__( |
| 21 | self, |
| 22 | dataset, |
| 23 | class_num, |
| 24 | image_mean, |
| 25 | image_std, |
| 26 | network, |
| 27 | multi_scales, |
| 28 | is_flip, |
| 29 | devices, |
| 30 | verbose=False, |
| 31 | save_path=None, |
| 32 | show_image=False, |
| 33 | ): |
| 34 | self.dataset = dataset |
| 35 | self.ndata = self.dataset.get_length() |
| 36 | self.class_num = class_num |
| 37 | self.image_mean = image_mean |
| 38 | self.image_std = image_std |
| 39 | self.multi_scales = multi_scales |
| 40 | self.is_flip = is_flip |
| 41 | self.network = network |
| 42 | self.devices = devices |
| 43 | |
| 44 | self.context = mp.get_context("spawn") |
| 45 | self.val_func = None |
| 46 | self.results_queue = self.context.Queue(self.ndata) |
| 47 | |
| 48 | self.verbose = verbose |
| 49 | self.save_path = save_path |
| 50 | if save_path is not None: |
| 51 | ensure_dir(save_path) |
| 52 | self.show_image = show_image |
| 53 | |
| 54 | def run(self, model_path, model_indice, log_file, log_file_link): |
| 55 | """There are four evaluation modes: |
nothing calls this directly
no test coverage detected