Index

_

__call__() (AstroPredictor method) (ImageReader method) (KRandomAugmentationList method) (test_mapper_cls method) (train_mapper_cls method) __repr__() (_TransformToAug method) (KRandomAugmentationList method) __str__ (_TransformToAug attribute) (KRandomAugmentationList attribute) _areaRng (COCOEvaluatorRecall attribute) _change_color_brightness() (Visualizer method) _convert_boxes() (Visualizer method) _convert_keypoints() (Visualizer method) _convert_masks() (Visualizer method) _cpu_device (COCOEvaluatorRecall attribute) _create_grayscale_image() (Visualizer method) _data_loader (LossEvalHook attribute) _default_font_size (Visualizer attribute) _derive_coco_results() (COCOEvaluatorRecall method) _distributed (COCOEvaluatorRecall attribute) _do_evaluation (COCOEvaluatorRecall attribute) _do_loss_eval() (LossEvalHook method) _draw_text_in_mask() (Visualizer method) _eval_predictions() (COCOEvaluatorRecall method) _evaluate_predictions_on_coco() (in module deepdisc.astrodet.astrodet) _forward_redshift() (CNNRedshiftPDFCasROIHeads method) (OldEBVRedshiftPDFCasROIHeads method) (OldRedshiftPDFCasROIHeads method) (RedshiftPDFCasROIHeads method) (RedshiftPDFCasROIHeadsGold method) (RedshiftPDFROIHeads method) (RedshiftPointCasROIHeads method) (RedshiftPointROIHeads method) (WeightedRedshiftPDFCasROIHeads method)

_get_loss() (LossEvalHook method) _instance_mode (Visualizer attribute) _jitter() (Visualizer method) _logger (COCOEvaluatorRecall attribute) _max_dets_per_image (COCOEvaluatorRecall attribute) _metadata (COCOEvaluatorRecall attribute) _model (LossEvalHook attribute) _optimizer (CustomLRScheduler attribute) _output_dir (COCOEvaluatorRecall attribute) _output_size (CNNRedshiftPDFCasROIHeads attribute) (OldEBVRedshiftPDFCasROIHeads attribute) (OldRedshiftPDFCasROIHeads attribute) (RedshiftPDFCasROIHeads attribute) (RedshiftPDFCasROIHeadsGold attribute) (RedshiftPDFROIHeads attribute) (RedshiftPointCasROIHeads attribute) (RedshiftPointROIHeads attribute) (WeightedRedshiftPDFCasROIHeads attribute) _period (LossEvalHook attribute) _plot_scene() (in module deepdisc.astrodet.scarlet_catalog) (in module deepdisc.astrodet.scarlet_nocatalog) (in module deepdisc.preprocessing.detection) _plot_wavelet() (in module deepdisc.astrodet.scarlet_catalog) (in module deepdisc.astrodet.scarlet_nocatalog) _read_image() (DC2ImageReader method) (HSCImageReader method) (ImageReader method) (RomanImageReader method) _scheduler (CustomLRScheduler attribute) _setup_augs() (KRandomAugmentationList method) _setup_figure() (VisImage method) _transform_to_aug() (in module deepdisc.astrodet.detectron) _TransformToAug (class in deepdisc.astrodet.detectron) _use_fast_impl (COCOEvaluatorRecall attribute) _verify_input_file_count() (DDLoader method)

A

A_EBV (in module deepdisc.data_format.augment_image) accumulate_custom() (COCOeval_opt_custom method) add_scaling() (ImageReader class method) add_val_loss() (LazyAstroTrainer method) (NewAstroTrainer method) addelementwise() (in module deepdisc.astrodet.astrodet) (in module deepdisc.data_format.augment_image) addelementwise16() (in module deepdisc.astrodet.astrodet) (in module deepdisc.data_format.augment_image) addelementwise8() (in module deepdisc.astrodet.astrodet) (in module deepdisc.data_format.augment_image) after_step() (CustomLRScheduler method) (LossEvalHook method)

after_train() (SaveHook method) annotate_dc2() (in module deepdisc.data_format.annotation_functions.annotate_dc2) annotate_dc2_wcs() (in module deepdisc.data_format.annotation_functions.annotate_dc2) annotate_decam() (in module deepdisc.data_format.annotation_functions.annotate_decam) annotate_hsc() (in module deepdisc.data_format.annotation_functions.annotate_hsc) apply_coords() (GenericWrapperTransform method) apply_image() (GenericWrapperTransform method) apply_segmentation() (GenericWrapperTransform method) AstroPredictor (class in deepdisc.astrodet.astrodet) aug (AstroPredictor attribute) augmentations (DataMapper attribute) augs (KRandomAugmentationList attribute)

B

before_train() (CustomLRScheduler method)

build_lr_scheduler() (LazyAstroTrainer class method) (NewAstroTrainer class method)

C

centercrop() (in module deepdisc.data_format.augment_image) cfg (AstroPredictor attribute) checkpointer (LazyAstroTrainer attribute) (NewAstroTrainer attribute) CNNRedshiftPDFCasROIHeads (class in deepdisc.model.models) COCOeval_opt_custom (class in deepdisc.astrodet.astrodet) COCOEvaluatorRecall (class in deepdisc.astrodet.astrodet) colormap() (in module deepdisc.astrodet.colormap)

ColorMode (class in deepdisc.astrodet.visualizer) convert_to_coco_dict() (in module deepdisc.astrodet.astrodet) convert_to_coco_json() (in module deepdisc.astrodet.astrodet) convert_to_json() (in module deepdisc.data_format.conversions) (in module deepdisc.data_format.file_io) cpu_device (Visualizer attribute) cropaug (KRandomAugmentationList attribute) CustomAug (class in deepdisc.astrodet.detectron) CustomLRScheduler (class in deepdisc.astrodet.detectron)

D

DataMapper (class in deepdisc.model.loaders) dataset (DDLoader attribute) dc2_train_augs() (in module deepdisc.data_format.augment_image) dc2_train_augs_full() (in module deepdisc.data_format.augment_image) DC2ImageReader (class in deepdisc.data_format.image_readers) ddict_to_hdf5() (in module deepdisc.data_format.conversions) DDLoader (class in deepdisc.data_format.file_io) deepdisc module deepdisc.astrodet module deepdisc.astrodet.astrodet module deepdisc.astrodet.colormap module deepdisc.astrodet.DC2_public module deepdisc.astrodet.detectron module deepdisc.astrodet.hsc module deepdisc.astrodet.scarlet_catalog module deepdisc.astrodet.scarlet_nocatalog module deepdisc.astrodet.visualizer module deepdisc.data_format module deepdisc.data_format.annotation_functions module deepdisc.data_format.annotation_functions.annotate_dc2 module deepdisc.data_format.annotation_functions.annotate_decam module deepdisc.data_format.annotation_functions.annotate_hsc module deepdisc.data_format.augment_image module deepdisc.data_format.conversions module deepdisc.data_format.file_io module deepdisc.data_format.image_readers module deepdisc.data_format.register_data module

deepdisc.inference module deepdisc.inference.match_objects module deepdisc.inference.predictors module deepdisc.model module deepdisc.model.loaders module deepdisc.model.models module deepdisc.preprocessing module deepdisc.preprocessing.detection module deepdisc.preprocessing.get_data module deepdisc.preprocessing.process module deepdisc.training module deepdisc.training.trainers module deepdisc.utils module deepdisc.utils.parse_arguments module default() (NpEncoder method) DictMapper (class in deepdisc.model.loaders) draw_and_connect_keypoints() (Visualizer method) draw_binary_mask() (Visualizer method) draw_box() (Visualizer method) draw_circle() (Visualizer method) draw_dataset_dict() (Visualizer method) draw_instance_predictions() (Visualizer method) draw_line() (Visualizer method) draw_panoptic_seg() (Visualizer method) draw_panoptic_seg_predictions (Visualizer attribute) draw_polygon() (Visualizer method) draw_rotated_box_with_label() (Visualizer method) draw_sem_seg() (Visualizer method) draw_soft_mask() (Visualizer method) draw_text() (Visualizer method) dtype_from_args() (in module deepdisc.utils.parse_arguments)

E

evaluate_custom() (COCOeval_opt_custom method)

F

filedict (DDLoader attribute) FILT_INX (in module deepdisc.data_format.annotation_functions.annotate_dc2) (in module deepdisc.data_format.annotation_functions.annotate_hsc) fit_scarlet_blend() (in module deepdisc.astrodet.scarlet_catalog) (in module deepdisc.astrodet.scarlet_nocatalog) (in module deepdisc.preprocessing.detection) fitsim_to_hdf5() (in module deepdisc.data_format.conversions) fitsim_to_numpy() (in module deepdisc.data_format.conversions)

forward() (CNNRedshiftPDFCasROIHeads method) (OldEBVRedshiftPDFCasROIHeads method) (OldRedshiftPDFCasROIHeads method) (RedshiftPDFCasROIHeads method) (RedshiftPDFCasROIHeadsGold method) (RedshiftPDFROIHeads method) (RedshiftPointCasROIHeads method) (RedshiftPointROIHeads method) (WeightedRedshiftPDFCasROIHeads method)

G

gaussblur() (in module deepdisc.astrodet.astrodet) (in module deepdisc.data_format.augment_image) generate_dataset_dict() (DDLoader method) generate_filedict() (DDLoader method) GenericWrapperTransform (class in deepdisc.astrodet.detectron) get_astro_dicts() (in module deepdisc.astrodet.astrodet) get_best_param_group_id() (CustomLRScheduler static method) get_centers() (in module deepdisc.astrodet.DC2_public) (in module deepdisc.preprocessing.get_data) get_cutout() (in module deepdisc.astrodet.DC2_public) (in module deepdisc.preprocessing.get_data) get_cutout_cat() (in module deepdisc.preprocessing.get_data) get_data_from_json() (in module deepdisc.data_format.file_io) get_dataset() (DDLoader method) get_DC2_data() (in module deepdisc.astrodet.DC2_public) (in module deepdisc.preprocessing.get_data) get_DC2_data_alltracts() (in module deepdisc.preprocessing.get_data) get_hsc_data() (in module deepdisc.astrodet.hsc)

get_hsc_DR3_data() (in module deepdisc.astrodet.hsc) get_image() (VisImage method) get_matched_object_classes() (in module deepdisc.inference.match_objects) get_matched_object_classes_new() (in module deepdisc.inference.match_objects) get_matched_object_inds() (in module deepdisc.inference.match_objects) get_matched_z_pdfs() (in module deepdisc.inference.match_objects) get_matched_z_pdfs_new() (in module deepdisc.inference.match_objects) get_matched_z_points_new() (in module deepdisc.inference.match_objects) get_object_coords() (in module deepdisc.inference.match_objects) get_output() (Visualizer method) get_predictions() (in module deepdisc.inference.predictors) get_predictions_new() (in module deepdisc.inference.predictors) get_processed_hsc_DR3_data() (in module deepdisc.astrodet.scarlet_catalog) (in module deepdisc.astrodet.scarlet_nocatalog) get_tract_patch_from_coord() (in module deepdisc.astrodet.hsc) get_transform() (_TransformToAug method) (CustomAug method) GoldRedshiftDictMapper (class in deepdisc.model.loaders) GoldRedshiftDictMapperEval (class in deepdisc.model.loaders)

H

hsc_test_augs() (in module deepdisc.data_format.augment_image)

HSCImageReader (class in deepdisc.data_format.image_readers)

I

IMAGE (ColorMode attribute) IMAGE_BW (ColorMode attribute) ImageReader (class in deepdisc.data_format.image_readers) img (VisImage attribute) (Visualizer attribute)

input_format (AstroPredictor attribute) inverse() (GenericWrapperTransform method) IR (DataMapper attribute) iterCount (LazyAstroTrainer attribute) (NewAstroTrainer attribute)

K

k (KRandomAugmentationList attribute) keypoint_threshold (Visualizer attribute)

km (DataMapper attribute) KRandomAugmentationList (class in deepdisc.astrodet.detectron)

L

LAMBDA_EFFS (in module deepdisc.data_format.augment_image) LazyAstroTrainer (class in deepdisc.training.trainers) load_coco_json_file() (DDLoader method) load_state_dict() (CustomLRScheduler method)

logger (in module deepdisc.data_format.file_io) LossEvalHook (class in deepdisc.astrodet.detectron) lossList (LazyAstroTrainer attribute) (NewAstroTrainer attribute) lupton() (ImageReader method)

M

mad_wavelet_own() (in module deepdisc.astrodet.scarlet_catalog) (in module deepdisc.astrodet.scarlet_nocatalog) (in module deepdisc.preprocessing.detection) MagRedshiftDictMapper (class in deepdisc.model.loaders) make_catalog() (in module deepdisc.astrodet.scarlet_catalog) (in module deepdisc.astrodet.scarlet_nocatalog) (in module deepdisc.preprocessing.detection) make_inference_arg_parser() (in module deepdisc.utils.parse_arguments) make_rail_informer_arg_parser() (in module deepdisc.utils.parse_arguments) make_training_arg_parser() (in module deepdisc.utils.parse_arguments) map_data() (DataMapper method) (DictMapper method) (GoldRedshiftDictMapper method) (GoldRedshiftDictMapperEval method) (MagRedshiftDictMapper method) (RedshiftDictMapper method) (RedshiftDictMapperEval method) (RedshiftEBVDictMapper method), [1] (WCSDictmapper method) max_range (KRandomAugmentationList attribute) metadata (Visualizer attribute) module deepdisc deepdisc.astrodet deepdisc.astrodet.astrodet deepdisc.astrodet.colormap deepdisc.astrodet.DC2_public deepdisc.astrodet.detectron deepdisc.astrodet.hsc deepdisc.astrodet.scarlet_catalog deepdisc.astrodet.scarlet_nocatalog deepdisc.astrodet.visualizer deepdisc.data_format deepdisc.data_format.annotation_functions deepdisc.data_format.annotation_functions.annotate_dc2 deepdisc.data_format.annotation_functions.annotate_decam deepdisc.data_format.annotation_functions.annotate_hsc deepdisc.data_format.augment_image deepdisc.data_format.conversions deepdisc.data_format.file_io deepdisc.data_format.image_readers deepdisc.data_format.register_data deepdisc.inference deepdisc.inference.match_objects deepdisc.inference.predictors deepdisc.model deepdisc.model.loaders deepdisc.model.models deepdisc.preprocessing deepdisc.preprocessing.detection deepdisc.preprocessing.get_data deepdisc.preprocessing.process deepdisc.training deepdisc.training.trainers deepdisc.utils deepdisc.utils.parse_arguments

multiband_gaussblur() (in module deepdisc.data_format.augment_image)

N

NewAstroTrainer (class in deepdisc.astrodet.astrodet) NORM (in module deepdisc.preprocessing.get_data) norm_dict (ImageReader attribute) NpEncoder (class in deepdisc.data_format.file_io) num_components (CNNRedshiftPDFCasROIHeads attribute) (OldEBVRedshiftPDFCasROIHeads attribute) (OldRedshiftPDFCasROIHeads attribute) (RedshiftPDFCasROIHeads attribute) (RedshiftPDFCasROIHeadsGold attribute) (RedshiftPDFROIHeads attribute) (WeightedRedshiftPDFCasROIHeads attribute)

numpyim_to_hdf5() (in module deepdisc.data_format.conversions)

O

OldEBVRedshiftPDFCasROIHeads (class in deepdisc.model.models) OldRedshiftPDFCasROIHeads (class in deepdisc.model.models) output (Visualizer attribute) output_name (SaveHook attribute) output_pdf() (CNNRedshiftPDFCasROIHeads method) (OldEBVRedshiftPDFCasROIHeads method) (OldRedshiftPDFCasROIHeads method) (RedshiftPDFCasROIHeads method) (RedshiftPDFCasROIHeadsGold method) (RedshiftPDFROIHeads method) (WeightedRedshiftPDFCasROIHeads method)

overlapped_slices() (in module deepdisc.astrodet.scarlet_catalog) (in module deepdisc.astrodet.scarlet_nocatalog) overlay_instances() (Visualizer method) overlay_rotated_instances() (Visualizer method)

P

period (LazyAstroTrainer attribute) (NewAstroTrainer attribute)

plot_stretch_Q() (in module deepdisc.astrodet.detectron) (in module deepdisc.astrodet.scarlet_catalog) (in module deepdisc.astrodet.scarlet_nocatalog)

Q

Q (in module deepdisc.preprocessing.get_data)

R

random_color() (in module deepdisc.astrodet.colormap) random_colors() (in module deepdisc.astrodet.colormap) random_sample() (DDLoader method) raw() (ImageReader method) read_image() (in module deepdisc.astrodet.hsc) read_image_decam() (in module deepdisc.astrodet.astrodet) read_image_hsc() (in module deepdisc.astrodet.astrodet) redden() (in module deepdisc.data_format.augment_image) redshift_conv (CNNRedshiftPDFCasROIHeads attribute) redshift_fc (OldEBVRedshiftPDFCasROIHeads attribute) (OldRedshiftPDFCasROIHeads attribute) (RedshiftPDFCasROIHeads attribute) (RedshiftPDFCasROIHeadsGold attribute) (RedshiftPDFROIHeads attribute) (RedshiftPointCasROIHeads attribute) (RedshiftPointROIHeads attribute) (WeightedRedshiftPDFCasROIHeads attribute) redshift_pooler (CNNRedshiftPDFCasROIHeads attribute) (OldEBVRedshiftPDFCasROIHeads attribute) (OldRedshiftPDFCasROIHeads attribute) (RedshiftPDFCasROIHeads attribute) (RedshiftPDFCasROIHeadsGold attribute) (RedshiftPDFROIHeads attribute) (RedshiftPointCasROIHeads attribute) (RedshiftPointROIHeads attribute) (WeightedRedshiftPDFCasROIHeads attribute) RedshiftDictMapper (class in deepdisc.model.loaders) RedshiftDictMapperEval (class in deepdisc.model.loaders) RedshiftEBVDictMapper (class in deepdisc.model.loaders), [1] RedshiftPDFCasROIHeads (class in deepdisc.model.models)

RedshiftPDFCasROIHeadsGold (class in deepdisc.model.models) RedshiftPDFROIHeads (class in deepdisc.model.models) RedshiftPointCasROIHeads (class in deepdisc.model.models) RedshiftPointROIHeads (class in deepdisc.model.models) register_data_set() (in module deepdisc.data_format.register_data) reset_image() (VisImage method) return_evallosshook() (in module deepdisc.training.trainers) return_lazy_model() (in module deepdisc.model.models) return_lazy_trainer() (in module deepdisc.training.trainers) return_model_objects() (in module deepdisc.astrodet.scarlet_catalog) (in module deepdisc.astrodet.scarlet_nocatalog) return_optimizer() (in module deepdisc.training.trainers) return_predictor_transformer() (in module deepdisc.inference.predictors) return_savehook() (in module deepdisc.training.trainers) return_schedulerhook() (in module deepdisc.training.trainers) return_spliced_sources() (in module deepdisc.astrodet.scarlet_catalog) (in module deepdisc.astrodet.scarlet_nocatalog) return_test_loader() (in module deepdisc.model.loaders) return_train_loader() (in module deepdisc.model.loaders) ria (test_mapper_cls attribute) (train_mapper_cls attribute) RomanImageReader (class in deepdisc.data_format.image_readers) run_batched_get_object_coords() (in module deepdisc.inference.match_objects) run_batched_match_class() (in module deepdisc.inference.match_objects) run_batched_match_redshift() (in module deepdisc.inference.match_objects) run_scarlet() (in module deepdisc.astrodet.scarlet_catalog) (in module deepdisc.astrodet.scarlet_nocatalog) (in module deepdisc.preprocessing.detection) run_step() (LazyAstroTrainer method) (NewAstroTrainer method)

S

save() (VisImage method) SaveHook (class in deepdisc.astrodet.detectron) scale (VisImage attribute) scale_psf() (in module deepdisc.data_format.augment_image) scalekwargs (ImageReader attribute) scaling (ImageReader attribute) scheduler (CustomLRScheduler property) (LazyAstroTrainer attribute) (NewAstroTrainer attribute)

SEGMENTATION (ColorMode attribute) set_mpl_style() (in module deepdisc.astrodet.astrodet) set_output_name() (SaveHook method) set_period() (LazyAstroTrainer method) (NewAstroTrainer method) state_dict() (CustomLRScheduler method) stretch (in module deepdisc.preprocessing.get_data) summarize_custom() (COCOeval_opt_custom method)

T

test_mapper_cls (class in deepdisc.astrodet.astrodet) tfm (_TransformToAug attribute)

train_augs() (in module deepdisc.data_format.augment_image) train_mapper_cls (class in deepdisc.astrodet.astrodet)

V

valloss (LazyAstroTrainer attribute) (NewAstroTrainer attribute) vallossList (LazyAstroTrainer attribute) (NewAstroTrainer attribute)

VisImage (class in deepdisc.astrodet.visualizer) Visualizer (class in deepdisc.astrodet.visualizer)

W

WCSDictmapper (class in deepdisc.model.loaders) WeightedRedshiftPDFCasROIHeads (class in deepdisc.model.models) write_scarlet_results() (in module deepdisc.astrodet.scarlet_catalog) (in module deepdisc.astrodet.scarlet_nocatalog) (in module deepdisc.preprocessing.process)

write_scarlet_results_nomodels() (in module deepdisc.preprocessing.process)

Z

zloss_factor (CNNRedshiftPDFCasROIHeads attribute) (OldEBVRedshiftPDFCasROIHeads attribute) (OldRedshiftPDFCasROIHeads attribute) (RedshiftPDFCasROIHeads attribute) (RedshiftPDFCasROIHeadsGold attribute) (RedshiftPDFROIHeads attribute) (RedshiftPointCasROIHeads attribute) (WeightedRedshiftPDFCasROIHeads attribute)

zscore() (ImageReader method)