Upload MRIBrainSequenceBERT

modeling.py

@@ -34,7 +34,7 @@ class MRIBrainSequenceBERT(PreTrainedModel):
             "ImageType",
             "Manufacturer",
             "ManufacturerModelName",
-            "ContrastBolusAgent",
+            # "ContrastBolusAgent",
             "ScanningSequence",
             "SequenceVariant",
             "ScanOptions",
@@ -55,8 +55,8 @@ class MRIBrainSequenceBERT(PreTrainedModel):
             "PercentSampling",
             "PercentPhaseFieldOfView",
             "PixelBandwidth",
-            "ContrastBolusVolume",
-            "ContrastBolusTotalDose",
+            # "ContrastBolusVolume",
+            # "ContrastBolusTotalDose",
             "AcquisitionMatrix",
             "InPlanePhaseEncodingDirection",
             "FlipAngle",
@@ -129,8 +129,11 @@ class MRIBrainSequenceBERT(PreTrainedModel):
         return x
 
     @staticmethod
-    def determine_plane_from_dicom(ds: pydicom.Dataset):
-        iop = np.asarray(ds.ImageOrientationPatient)
+    def determine_plane_from_dicom(ds: pydicom.Dataset | dict):
+        iop = ds.get("ImageOrientationPatient", None)
+        if iop is None:
+            return None
+        iop = np.asarray(iop)
         # Calculate the direction cosine for the normal vector of the plane
         normal_vector = np.cross(iop[:3], iop[3:])