deepdisc.inference.match_objects

Functions

get_matched_object_inds(dataset_dict, outputs)	Returns indices for matched pairs of ground truth and detected objects in an image
get_object_coords(dataset_dict, outputs)	Returns indices for matched pairs of ground truth and detected objects in an image
get_matched_object_classes(dataset_dicts, imreader, ...)	Returns object classes for matched pairs of ground truth and detected objects in an image
get_matched_object_classes_new(dataset_dicts, predictor)	Returns object classes for matched pairs of ground truth and detected objects test images
get_matched_z_pdfs(dataset_dicts, imreader, ...[, ...])	Returns redshift pdfs for matched pairs of ground truth and detected objects test images
get_matched_z_pdfs_new(dataset_dicts, predictor[, ...])	Returns redshift pdfs for matched pairs of ground truth and detected objects test images
get_matched_z_points_new(dataset_dicts, predictor)	Returns redshift point estimates for matched pairs of ground truth and detected objects test images
run_batched_match_class(dataloader, predictor)	Test function not yet implemented for batch prediction
run_batched_match_redshift(dataloader, predictor[, ...])	Test function not yet implemented for batch prediction
run_batched_get_object_coords(dataloader, predictor)	Test function not yet implemented for batch prediction

Module Contents

get_matched_object_inds(dataset_dict, outputs)

Returns indices for matched pairs of ground truth and detected objects in an image

Parameters:

dataset_dict (dictionary) – The dictionary metadata for a single image

Returns:

• matched_gts (list(int)) – The indices of matched objects in the ground truth list

• matched_dts (list(int)) – The indices of matched objects in the detections list

• outputs (list(Intances)) – The list of detected object Instances

get_object_coords(dataset_dict, outputs)

Returns indices for matched pairs of ground truth and detected objects in an image

Parameters:

dataset_dict (dictionary) – The dictionary metadata containing the wcs for a single image

Returns:

• matched_gts (list(int)) – The indices of matched objects in the ground truth list

• matched_dts (list(int)) – The indices of matched objects in the detections list

• outputs (list(Intances)) – The list of detected object Instances

get_matched_object_classes(dataset_dicts, imreader, key_mapper, predictor)

Returns object classes for matched pairs of ground truth and detected objects in an image

Parameters:

• dataset_dicts (list[dict]) – The dictionary metadata for a test images

• imreader (ImageReader object) – An object derived from ImageReader base class to read the images.

• key_mapper (function) – The key_mapper should take a dataset_dict as input and return the key used by imreader

• predictor (AstroPredictor) – The predictor object used to make predictions on the test set

Returns:

• true_classes (list(int)) – The classes of matched objects in the ground truth list

• pred_classes (list(int)) – The classes of matched objects in the detections list

get_matched_object_classes_new(dataset_dicts, predictor)

Returns object classes for matched pairs of ground truth and detected objects test images assuming the dataset_dicts have the image HxWxC in the ‘image_shaped’ field

Parameters:

• dataset_dicts (list[dict]) – The dictionary metadata for a test images

• predictor (AstroPredictor) – The predictor object used to make predictions on the test set

Returns:

• true_classes (list(int)) – The classes of matched objects in the ground truth list

• pred_classes (list(int)) – The classes of matched objects in the detections list

get_matched_z_pdfs(dataset_dicts, imreader, key_mapper, predictor, ids=False, blendedness=False)

Returns redshift pdfs for matched pairs of ground truth and detected objects test images

Parameters:

• dataset_dicts (list[dict]) – The dictionary metadata for a test images

• imreader (ImageReader object) – An object derived from ImageReader base class to read the images.

• key_mapper (function) – The key_mapper should take a dataset_dict as input and return the key used by imreader

• predictor (AstroPredictor) – The predictor object used to make predictions on the test set

Returns:

• z_trues (list(float)) – The redshifts of matched objects in the ground truth list

• z_preds (list(array(float))) – The redshift pdfs of matched objects in the detections list

get_matched_z_pdfs_new(dataset_dicts, predictor, ids=False, blendedness=False)

Returns redshift pdfs for matched pairs of ground truth and detected objects test images assuming the dataset_dicts have the image HxWxC in the ‘image_shaped’ field

Parameters:

• dataset_dicts (list[dict]) – The dictionary metadata for a test images

• predictor (AstroPredictor) – The predictor object used to make predictions on the test set

Returns:

• z_trues (list(float)) – The redshifts of matched objects in the ground truth list

• z_preds (list(array(float))) – The redshift pdfs of matched objects in the detections list

get_matched_z_points_new(dataset_dicts, predictor)

Returns redshift point estimates for matched pairs of ground truth and detected objects test images assuming the dataset_dicts have the image in the ‘image_shaped’ field

Parameters:

• dataset_dicts (list[dict]) – The dictionary metadata for a test images

• predictor (AstroPredictor) – The predictor object used to make predictions on the test set

Returns:

• z_trues (list(float)) – The redshifts of matched objects in the ground truth list

• z_preds (list(array(float))) – The redshift pdfs of matched objects in the detections list

run_batched_match_class(dataloader, predictor)

Test function not yet implemented for batch prediction

run_batched_match_redshift(dataloader, predictor, ids=False, blendedness=False)

Test function not yet implemented for batch prediction

run_batched_get_object_coords(dataloader, predictor)

Test function not yet implemented for batch prediction