Import pydicom more robustly

lishen · Dec 7, 2019 · e95c1f9 · e95c1f9
1 parent d4157e2
commit e95c1f9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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()