NeuralNetInferenceClient: Accept PixelMask

This commit is contained in:
2025-06-30 21:29:48 +02:00
parent dba2544c48
commit 92288c60d7
7 changed files with 110 additions and 37 deletions
+3 -2
View File
@@ -18,7 +18,8 @@ ImageAnalysisCPU::ImageAnalysisCPU(const DiffractionExperiment &in_experiment,
mask_1byte(npixels, 0),
spotFinder(in_integration),
saturation_limit(experiment.GetSaturationLimit()),
roi_count(0) {
roi_count(0),
mask(in_mask) {
nquads = 2;
@@ -180,7 +181,7 @@ void ImageAnalysisCPU::Analyze(DataMessage &output,
output.az_int_profile = profile.GetResult();
output.bkg_estimate = profile.GetBkgEstimate(integration.Settings());
if ((inference_client != nullptr) && spot_finding_settings.resolution_estimate)
output.resolution_estimate = inference_client->Inference(experiment, image, nquads);
output.resolution_estimate = inference_client->Inference(experiment, mask, image, nquads);
for (const auto &[key, val]: roi_names)
output.roi[key] = roi[val];