|
21 | 21 | import numpy as np |
22 | 22 | import pydicom |
23 | 23 | import SimpleITK |
| 24 | +from monai.transforms import LoadImage |
24 | 25 | from pydicom.filereader import dcmread |
25 | 26 | from pydicom.sr.codedict import codes |
26 | 27 |
|
|
37 | 38 | from monailabel import __version__ |
38 | 39 | from monailabel.config import settings |
39 | 40 | from monailabel.datastore.utils.colors import GENERIC_ANATOMY_COLORS |
| 41 | +from monailabel.transform.writer import write_itk |
40 | 42 |
|
41 | 43 | logger = logging.getLogger(__name__) |
42 | 44 |
|
@@ -208,12 +210,10 @@ def dicom_to_nifti(series_dir, is_seg=False): |
208 | 210 | logger.info(f"dicom_to_nifti: Converting DICOM from {series_dir} using NvDicomReader") |
209 | 211 |
|
210 | 212 | try: |
211 | | - from monai.transforms import LoadImage |
212 | 213 | from monailabel.transform.reader import NvDicomReader |
213 | | - from monailabel.transform.writer import write_itk |
214 | 214 |
|
215 | 215 | # Use NvDicomReader with LoadImage |
216 | | - reader = NvDicomReader(reverse_indexing=True, use_nvimgcodec=True) |
| 216 | + reader = NvDicomReader(reverse_indexing=True) |
217 | 217 | loader = LoadImage(reader=reader, image_only=False) |
218 | 218 |
|
219 | 219 | # Load the DICOM (supports both directories and single files) |
@@ -867,7 +867,7 @@ def transcode_dicom_to_htj2k( |
867 | 867 | if verify: |
868 | 868 | ds_verify = pydicom.dcmread(output_file) |
869 | 869 | pixel_data = ds_verify.PixelData |
870 | | - data_sequence = pydicom.encaps.decode_data_sequence(pixel_data) |
| 870 | + data_sequence = [fragment for fragment in pydicom.encaps.generate_frames(pixel_data)] |
871 | 871 | images_verify = decoder.decode( |
872 | 872 | data_sequence, |
873 | 873 | params=nvimgcodec.DecodeParams( |
|
0 commit comments