Skip to content

Commit

Permalink
update README
Browse files Browse the repository at this point in the history
  • Loading branch information
YijinHuang committed Jul 17, 2021
1 parent cf3d33c commit 04b3b03
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -49,27 +49,27 @@ $ pip install -r requirements.txt

1. Download EyePACS dataset. Then run `tools/crop.py` to remove the black border of images and resize them to 512 x 512.
2. Rename all images as 'id_eyeSide.jpeg', where 'id' here is the id of images given by EyePACS and 'eyeSide' is left or right. Then move all images into a folder.
3. Download the provided lesion predicted result, which is a pickle file containing a dict as follows:
3. Download the provided [lesion predictions](https://github.com/YijinHuang/Lesion-based-Contrastive-Learning/releases/tag/v1.0), which is a pickle file containing a dict as follows:

```python
predictions = {
'train': {
'id_eyeSide.jpeg': [(x1, y1, x2, y2), ..., (x1, y1, x2, y2)],
'id_eyeSide.jpeg': [(x1, y1, x2, y2), ..., (x1, y1, x2, y2)],
'id_eyeSide.jpeg': [(x1, y1, x2, y2), ..., (x1, y1, x2, y2)],
'{id}_{eyeSide}.jpeg': [(x1, y1, x2, y2), ..., (x1, y1, x2, y2)],
'{id}_{eyeSide}.jpeg': [(x1, y1, x2, y2), ..., (x1, y1, x2, y2)],
'{id}_{eyeSide}.jpeg': [(x1, y1, x2, y2), ..., (x1, y1, x2, y2)],
...
},
'val': {
'id_eyeSide.jpeg': [(x1, y1, x2, y2), ..., (x1, y1, x2, y2)],
'id_eyeSide.jpeg': [(x1, y1, x2, y2), ..., (x1, y1, x2, y2)],
'{id}_{eyeSide}.jpeg': [(x1, y1, x2, y2), ..., (x1, y1, x2, y2)],
'{id}_{eyeSide}.jpeg': [(x1, y1, x2, y2), ..., (x1, y1, x2, y2)],
...
}
}
```

#### Training

1. Update 'data_path' and 'data_index' in `config.py`, where 'data_path' is the folder containing preprocessed images and 'data_index' is the pickle file with lesion predicted results. You can update other training configurations and hyperparameters in `config.py` for your customized dataset.
1. Update 'data_path' and 'data_index' in `config.py`, where 'data_path' is the folder containing preprocessed images and 'data_index' is the pickle file with lesion predicted results. You can update other training configurations and hyper-parameters in `config.py` for your customized dataset.
2. Run to train:

```shell
Expand Down

0 comments on commit 04b3b03

Please sign in to comment.