Spaces:
Sleeping
Sleeping
| import pydicom | |
| import numpy as np | |
| from skimage.transform import resize | |
| def load_dicom(file_path): | |
| """Load a DICOM file and return it as a numpy array.""" | |
| dicom = pydicom.dcmread(file_path) | |
| return dicom.pixel_array | |
| def preprocess_dicom(dicom_array, target_size=(256, 256)): | |
| """Preprocess the DICOM array by resizing and normalizing.""" | |
| # Resize the image | |
| resized = resize(dicom_array, target_size, mode='constant', preserve_range=True) | |
| # Normalize the image | |
| normalized = (resized - np.min(resized)) / (np.max(resized) - np.min(resized)) | |
| return normalized.astype(np.float32) | |
| def load_and_preprocess_dicom(file_path, target_size=(256, 256)): | |
| """Load a DICOM file, preprocess it, and return as a numpy array.""" | |
| dicom_array = load_dicom(file_path) | |
| return preprocess_dicom(dicom_array, target_size) |