[GCOLAB] Added DICOMLoader

YanSte · Aug 29, 2023 · 06e6a86 · 06e6a86
1 parent 0b34ccb
commit 06e6a86
Show file tree

Hide file tree

Showing 3 changed files with 432 additions and 5 deletions.
diff --git a/src/skit/config.py b/src/skit/config.py
@@ -9,7 +9,6 @@
 except ImportError:
     IN_COLAB = False
 
-
 # ==============================
 #           Library
 # ==============================
@@ -20,3 +19,10 @@
     IS_TENSORFLOW_IMPORTED = True
 except ImportError:
     IS_TENSORFLOW_IMPORTED = False
+
+# Env with library tensorflow
+try:
+    import pydicom
+    IS_PYDICOM_IMPORTED = True
+except ImportError:
+    IS_PYDICOM_IMPORTED = False