ianpan commited on
Commit
bec2e30
·
verified ·
1 Parent(s): 79ef347

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -44,7 +44,9 @@ def crop_roi(img):
44
  ```
45
 
46
  While not guaranteed to be foolproof, a cursory review of a sample of cropped images demonstrated excellent performance.
47
- The model was trained with a large batch size (256) to mitigate noise.
 
 
48
 
49
  To use the model:
50
  ```
 
44
  ```
45
 
46
  While not guaranteed to be foolproof, a cursory review of a sample of cropped images demonstrated excellent performance.
47
+ The model was trained with a larger batch size (256) to mitigate noise. Input into the model is expected to be [0, 255].
48
+ If you are loading from DICOM, you should convert the input into an 8-bit image, pass it through `model.preprocess`, and
49
+ ensure that it is a float `torch.Tensor`before passing it to the model. The normalization step happens within the model itself.
50
 
51
  To use the model:
52
  ```