core.BasePredictor¶

Module Contents¶

class core.BasePredictor.BasePredictor(model, device, input_path, output_path, make_submission=False, exp_name=None, algo_params=None)[source]¶

model[source]¶

device[source]¶

input_path[source]¶

output_path[source]¶

make_submission = False[source]¶

exp_name = None[source]¶

conduct_prediction()[source]¶

write_pred_mask(pred_mask, output_dir, image_name, submission=False)[source]¶