o # i@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~mZmZddlmZmZmZddlmZmZmZmZmZeeddlmZddZddZeePeWdn 1smwYdd 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)!affine_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)hasattrrvrxset_ requires_grad__dict__) clsstoragerwmetadatastorage_offsetsizestrider{ meta_dicttrU/home/dell461/cl/sdc2/last_ska_mid/HISourceFinder-master-l/src/monai/data/__init__.py _rebuild_meta|srcCsvt|dr |}nt|dr|}n|}|j}|jr"td||| |j |j f}t t ||||ffS)Nuntyped_storage_typed_storagez8sharing CUDA metatensor across processes not implemented)ryrrr~rwis_cudaNotImplementedErrorrrrr{r|rtype) meta_tensorr~rwrrrrreduce_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`rarbrcrdrerfrgrhrirjrkrlZ wsi_datasetsrmrnroZ wsi_readerrprqrrrsrtsuppress BaseExceptionZmultiprocessing.reductionrurrregisterZultrasound_confidence_maprrrrrs@ , 8  $(   &   !