Update README.md
Browse files
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
|
|
|
|
|
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 |
```
|