From e95c1f9d9a253724dc5edb1309447cc4fc563a62 Mon Sep 17 00:00:00 2001 From: Li Shen Date: Fri, 6 Dec 2019 23:08:15 -0500 Subject: [PATCH] Import pydicom more robustly --- dm_image.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/dm_image.py b/dm_image.py index ba1d402..e975a05 100644 --- a/dm_image.py +++ b/dm_image.py @@ -10,7 +10,10 @@ from keras.utils.np_utils import to_categorical import keras.backend as K import cv2 -import dicom +try: + import dicom +except ImportError: + import pydicom as dicom from dm_preprocess import DMImagePreprocessor as prep data_format = K.image_data_format()