Spaces:
Sleeping
Sleeping
File size: 3,212 Bytes
60a7004 c807f2c 60a7004 c807f2c 60a7004 c807f2c 60a7004 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 |
import os
import tempfile
import pydicom
import nibabel as nib
import numpy as np
import matplotlib.pyplot as plt
import gradio as gr
def dicom_to_nifti_and_display(dicom_file):
"""
Converte um arquivo DICOM para NIfTI e exibe as imagens antes e depois da conversão.
:param dicom_file: Arquivo DICOM carregado pelo usuário.
:return: Tupla contendo a imagem DICOM original e a imagem NIfTI convertida.
"""
try:
# Ler o arquivo DICOM
dicom_data = pydicom.dcmread(dicom_file.name)
dicom_image = dicom_data.pixel_array
# Verificar se a imagem é 3D e extrair o slice central
if len(dicom_image.shape) == 3:
slice_index = dicom_image.shape[0] // 2 # Slice central ao longo do eixo Z
dicom_slice = dicom_image[slice_index, :, :] # Extrair o slice
else:
dicom_slice = dicom_image # Se for 2D, use a imagem diretamente
# Criar um objeto NIfTI
nifti_image = nib.Nifti1Image(dicom_image, affine=np.eye(4))
# Salvar temporariamente o arquivo NIfTI
with tempfile.NamedTemporaryFile(suffix=".nii", delete=False) as temp_nifti:
nib.save(nifti_image, temp_nifti.name)
nifti_path = temp_nifti.name
# Carregar o arquivo NIfTI para exibição
nifti_loaded = nib.load(nifti_path).get_fdata()
# Exibir as imagens usando Matplotlib
fig, axes = plt.subplots(1, 2, figsize=(10, 5))
# Imagem DICOM (slice central)
axes[0].imshow(dicom_slice, cmap="gray")
axes[0].set_title("Imagem DICOM Original")
axes[0].axis("off")
# Imagem NIfTI (slice central)
if len(nifti_loaded.shape) == 3:
nifti_slice = nifti_loaded[nifti_loaded.shape[2] // 2, :, :] # Slice central ao longo do eixo Z
else:
nifti_slice = nifti_loaded
axes[1].imshow(nifti_slice, cmap="gray")
axes[1].set_title("Imagem NIfTI Convertida")
axes[1].axis("off")
# Salvar a figura em um buffer temporário
with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_image:
plt.savefig(temp_image.name, bbox_inches="tight")
plt.close(fig)
return temp_image.name, nifti_path
except Exception as e:
raise gr.Error(f"Erro ao processar o arquivo: {e}")
# Interface Gradio
with gr.Blocks() as demo:
gr.Markdown("# Conversor DICOM para NIfTI com Visualização")
gr.Markdown("Faça upload de um arquivo DICOM, visualize a imagem original, converta para NIfTI e veja o resultado.")
with gr.Row():
input_dicom = gr.File(label="Upload de Arquivo DICOM")
output_images = gr.Image(label="Imagens DICOM e NIfTI")
output_nifti = gr.File(label="Download do Arquivo NIfTI")
convert_button = gr.Button("Converter e Visualizar")
def process_and_display(dicom_file):
image_path, nifti_path = dicom_to_nifti_and_display(dicom_file)
return image_path, nifti_path
convert_button.click(
process_and_display,
inputs=input_dicom,
outputs=[output_images, output_nifti]
)
# Executar a interface
demo.launch() |