o
    #i                     @  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mZ ddlmZmZmZ ddlmZmZmZmZmZ ee d dlmZ dd Zdd ZeePe W d   n	1 smw   Y  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)!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)hasattrrv   rx   set_requires_grad__dict__)
clsstoragerw   metadatastorage_offsetsizestrider{   	meta_dictt r   U/home/dell461/cl/sdc2/last_ska_mid/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r"td|  |  | 	 | j
| jf}tt| |||ffS )Nuntyped_storage_typed_storagez8sharing CUDA metatensor across processes not implemented)ry   r   r   r~   rw   is_cudaNotImplementedErrorr   r   r   r{   r|   r   type)meta_tensorr~   rw   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   Zwsi_datasetsrm   rn   ro   Z
wsi_readerrp   rq   rr   rs   rt   suppressBaseExceptionZmultiprocessing.reductionru   r   r   registerZultrasound_confidence_mapr   r   r   r   r   <module>   s@   ,8$(
 &	!