U
    Ph/                  	   @  s  d dl mZ d dlZddlmZmZmZmZmZm	Z	m
Z
mZmZ ddlmZ ddlmZ ddlmZmZmZmZmZmZmZmZmZmZmZmZ ddlmZ dd	l m!Z!m"Z"m#Z#m$Z$ dd
l%m&Z&m'Z' ddl(m)Z)m*Z*m+Z+m,Z, ddl-m.Z. ddl/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6 ddl7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z? ddl@mAZAmBZBmCZC ddlDmEZEmFZFmGZGmHZHmIZImJZJ ddlKmLZLmMZMmNZN ddlOmPZP ddlQmRZRmSZS ddlTmUZUmVZV ddlWmXZX ddlYmZZZm[Z[ ddl\m]Z]m^Z^ ddl_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZmZ ddlmZmZmZ ddlmZmZmZmZmZ ee. d dlmZ dd Zdd ZeePe W 5 Q R X dd lmZ dS )!    )annotationsN   )	box_areabox_centersbox_gioubox_ioubox_pair_giouboxes_center_distancecenters_in_boxesconvert_box_modeconvert_box_to_standard_mode)CSVSaver)
DataLoader)ArrayDatasetCacheDatasetCacheNTransDataset
CSVDatasetDatasetDatasetFunc
GDSDatasetLMDBDatasetNPZDictItemDatasetPersistentDatasetSmartCacheDataset
ZipDataset)DatasetSummary)check_missing_files create_cross_validation_datalistload_decathlon_datalistload_decathlon_properties)FolderLayoutFolderLayoutBase)GridPatchDatasetPatchDataset	PatchIter
PatchIterd)ImageDataset)ImageReader	ITKReaderNibabelReader
NrrdReaderNumpyReader	PILReaderPydicomReader)SUPPORTED_WRITERSImageWriter	ITKWriterNibabelWriter	PILWriterloggerregister_writerresolve_writer)CSVIterableDatasetIterableDatasetShuffleBuffer)get_itk_image_centeritk_image_to_metatensoritk_to_monai_affinemetatensor_to_itk_imagemonai_to_itk_affinemonai_to_itk_ddf)MetaObjget_track_metaset_track_meta)
MetaTensor)DistributedSampler DistributedWeightedRandomSampler)create_test_image_2dcreate_test_image_3d)TestTimeAugmentation)ThreadBufferThreadDataLoader)load_net_with_metadatasave_net_with_metadata)"PICKLE_KEY_SUFFIXaffine_to_spacingcompute_importance_mapcompute_shape_offsetconvert_tables_to_dicts!correct_nifti_header_if_necessarycreate_file_basenamedecollate_batchdense_patch_slicesget_extra_metadata_keysget_random_patchget_valid_patch_sizeis_supported_format
iter_patchiter_patch_positioniter_patch_slicesjson_hashinglist_data_collateorientation_ras_lpspad_list_data_collatepartition_datasetpartition_dataset_classespickle_hashingrectify_header_sform_qformremove_extra_metadataremove_keysreorient_spatial_axesresample_datalistselect_cross_validation_foldsset_rndsorted_dictto_affine_ndworker_init_fnzoom_affine)MaskedPatchWSIDatasetPatchWSIDatasetSlidingPatchWSIDataset)BaseWSIReaderCuCIMWSIReaderOpenSlideWSIReaderTiffFileWSIReader	WSIReader)ForkingPicklerc           
      C  sR   |\}}}}}t |dr|jn|}| g ||jd}	|	|||| ||	_||	_|	S )N_untyped_storage)dtypedevice)hasattrrw   ry   set_requires_grad__dict__)
clsstoragerx   metadatastorage_offsetsizestrider|   	meta_dictt r   H/home/dell461/cl/sdc2/HISourceFinder-master-l/src/monai/data/__init__.py_rebuild_meta}   s    r   c                 C  sv   t | dr|  }nt | dr(|  }n|  }| j}| jrDtd|  |  | 	 | j
| jf}tt| |||ffS )Nuntyped_storage_typed_storagez8sharing CUDA metatensor across processes not implemented)rz   r   r   r   rx   is_cudaNotImplementedErrorr   r   r   r|   r}   r   type)meta_tensorr   rx   r   r   r   r   reduce_meta_tensor   s    



r   )UltrasoundConfidenceMap)
__future__r   
contextlibZ	box_utilsr   r   r   r   r   r	   r
   r   r   Z	csv_saverr   
dataloaderr   datasetr   r   r   r   r   r   r   r   r   r   r   r   Zdataset_summaryr   Zdecathlon_datalistr   r   r   r   Z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/   r0   r1   r2   r3   r4   r5   Ziterable_datasetr6   r7   r8   Zitk_torch_bridger9   r:   r;   r<   r=   r>   Zmeta_objr?   r@   rA   r   rB   ZsamplersrC   rD   Z	syntheticrE   rF   Ztest_time_augmentationrG   Zthread_bufferrH   rI   Ztorchscript_utilsrJ   rK   utilsrL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   Zwsi_datasetsrn   ro   rp   Z
wsi_readerrq   rr   rs   rt   ru   suppressBaseExceptionZmultiprocessing.reductionrv   r   r   registerZultrasound_confidence_mapr   r   r   r   r   <module>   s<   ,8$(
 '	