Skip to content
This repository has been archived by the owner on Jul 16, 2024. It is now read-only.

Commit

Permalink
Browse files Browse the repository at this point in the history
  • Loading branch information
laculp committed May 11, 2023
2 parents 2e4a594 + 481c022 commit ca00382
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ For contrastive pretraining, we build on [SimCLR](https://github.com/google-rese

In our default contrastive pretraining setting, we utilized random cropping (C), random color distortion (D), rotation (R), and random Gaussian blur (G) as the data augmentation strategy. Due to the grayscale nature of radiology images, for these images we opted for stronger data augmentation to reduce the chances of overfitting. We further improved the final performance by incorporating histogram equalization and elastic deformation in addition to our default data augmentation strategy.


### **Training Data**
We open-source models trained on public medical data only. This is available for chest x-ray and pathology only. The data used in each model are the following:

Expand Down Expand Up @@ -93,7 +94,6 @@ wget -r -N -c -np --user <physionet-username> --ask-password https://physionet.o
The Tensorflow 2 saved model format can be loaded as follows. See further information about hub Module [here](https://www.tensorflow.org/hub/api_docs/python/hub/Module).

```
import tensorflow_hub as hub
module = hub.load('TOP_LEVEL_HUB_PATH')
Expand All @@ -103,7 +103,6 @@ module = hub.load('TOP_LEVEL_HUB_PATH')
image = <LOAD_IMAGE_HERE>
embedding_of_image = module(image)
```


Expand Down

0 comments on commit ca00382

Please sign in to comment.