U Ph/ @sddlmZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZddlmZdd l m!Z!m"Z"m#Z#m$Z$dd l%m&Z&m'Z'dd l(m)Z)m*Z*m+Z+m,Z,dd l-m.Z.dd l/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6ddl7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?ddl@mAZAmBZBmCZCddlDmEZEmFZFmGZGmHZHmIZImJZJddlKmLZLmMZMmNZNddlOmPZPddlQmRZRmSZSddlTmUZUmVZVddlWmXZXddlYmZZZm[Z[ddl\m]Z]m^Z^ddl_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZmZddlmZmZmZddlmZmZmZmZmZee.ddlmZddZddZeePeW5QRXdd lmZdS)!) annotationsN) box_area box_centersbox_gioubox_iou box_pair_giouboxes_center_distancecenters_in_boxesconvert_box_modeconvert_box_to_standard_mode)CSVSaver) DataLoader) ArrayDataset CacheDatasetCacheNTransDataset CSVDatasetDataset DatasetFunc GDSDataset LMDBDatasetNPZDictItemDatasetPersistentDatasetSmartCacheDataset ZipDataset)DatasetSummary)check_missing_files create_cross_validation_datalistload_decathlon_datalistload_decathlon_properties) FolderLayoutFolderLayoutBase)GridPatchDataset PatchDataset PatchIter PatchIterd) ImageDataset) ImageReader ITKReader NibabelReader NrrdReader NumpyReader PILReader PydicomReader)SUPPORTED_WRITERS ImageWriter ITKWriter NibabelWriter PILWriterloggerregister_writerresolve_writer)CSVIterableDatasetIterableDataset ShuffleBuffer)get_itk_image_centeritk_image_to_metatensoritk_to_monai_affinemetatensor_to_itk_imagemonai_to_itk_affinemonai_to_itk_ddf)MetaObjget_track_metaset_track_meta) MetaTensor)DistributedSampler DistributedWeightedRandomSampler)create_test_image_2dcreate_test_image_3d)TestTimeAugmentation) ThreadBufferThreadDataLoader)load_net_with_metadatasave_net_with_metadata)"PICKLE_KEY_SUFFIXaffine_to_spacingcompute_importance_mapcompute_shape_offsetconvert_tables_to_dicts!correct_nifti_header_if_necessarycreate_file_basenamedecollate_batchdense_patch_slicesget_extra_metadata_keysget_random_patchget_valid_patch_sizeis_supported_format iter_patchiter_patch_positioniter_patch_slices json_hashinglist_data_collateorientation_ras_lpspad_list_data_collatepartition_datasetpartition_dataset_classespickle_hashingrectify_header_sform_qformremove_extra_metadata remove_keysreorient_spatial_axesresample_datalistselect_cross_validation_foldsset_rnd sorted_dict to_affine_ndworker_init_fn zoom_affine)MaskedPatchWSIDatasetPatchWSIDatasetSlidingPatchWSIDataset) BaseWSIReaderCuCIMWSIReaderOpenSlideWSIReaderTiffFileWSIReader WSIReader)ForkingPicklerc CsR|\}}}}}t|dr|jn|}|g||jd} | |||||| _|| _| S)N_untyped_storage)dtypedevice)hasattrrwryset_ requires_grad__dict__) clsstoragerxmetadatastorage_offsetsizestrider| meta_dicttrH/home/dell461/cl/sdc2/HISourceFinder-master-l/src/monai/data/__init__.py _rebuild_meta}srcCsvt|dr|}nt|dr(|}n|}|j}|jrDtd||| |j |j f}t t ||||ffS)Nuntyped_storage_typed_storagez8sharing CUDA metatensor across processes not implemented)rzrrrrxis_cudaNotImplementedErrorrrrr|r}rtype) meta_tensorrrxrrrrreduce_meta_tensors    r)UltrasoundConfidenceMap) __future__r contextlibZ box_utilsrrrrrr r r r Z csv_saverr dataloaderrdatasetrrrrrrrrrrrrZdataset_summaryrZdecathlon_datalistrrrrZ folder_layoutr r!Z grid_datasetr"r#r$r%Z image_datasetr&Z image_readerr'r(r)r*r+r,r-Z image_writerr.r/r0r1r2r3r4r5Ziterable_datasetr6r7r8Zitk_torch_bridger9r:r;r<r=r>Zmeta_objr?r@rArrBZsamplersrCrDZ syntheticrErFZtest_time_augmentationrGZ thread_bufferrHrIZtorchscript_utilsrJrKutilsrLrMrNrOrPrQrRrSrTrUrVrWrXrYrZr[r\r]r^r_r`rarbrcrdrerfrgrhrirjrkrlrmZ wsi_datasetsrnrorpZ wsi_readerrqrrrsrtrusuppress BaseExceptionZmultiprocessing.reductionrvrrregisterZultrasound_confidence_maprrrrr s< ,  8  $(    '