fomofo
/

tap-ct-s-3d

Model card Files Files and versions

Update README.md

#1

by kbwgrootlipman - opened Jan 7

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -70,7 +70,7 @@ volume = sitk.DICOMOrient(volume, 'LPS')
 # Get array, expand to (B, C, D, H, W) and preprocess
 array = sitk.GetArrayFromImage(volume)
-array = np.expand_dims(array, axis(0, 1))
 x = preprocessor(array)['pixel_values']
 # Forward pass

 # Get array, expand to (B, C, D, H, W) and preprocess
 array = sitk.GetArrayFromImage(volume)
+array = np.expand_dims(array, axis=(0, 1))
 x = preprocessor(array)['pixel_values']
 # Forward pass