Skip to content

Commit

Permalink
[update] dataset preprocess usage
Browse files Browse the repository at this point in the history
  • Loading branch information
dengxl0520 committed Jun 11, 2024
1 parent e675d1f commit 02328a9
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,15 @@ First, download the dataset from:
- [CAMUS](https://www.creatis.insa-lyon.fr/Challenge/camus/index.html)
- [EchoNet-Dynamic](https://echonet.github.io/dynamic/index.html)

Then process the dataset according to `utils/preprocess_echonet.py` and `utils/preprocess_camus.py`.
Then process the dataset according to `utils/preprocess_echonet.py` and `utils/preprocess_camus.py`, for example:

```
# CAMUS
python utils/preprocess_camus.py -i /data/dengxiaolong/CAMUS_public/database_nifti -o /data/dengxiaolong/memsam/CAMUS_public
# EchoNet-Dynamic
python utils/preprocess_echonet.py -i /data/dengxiaolong/EchoNet-Dynamic -o /data/dengxiaolong/memsam/EchoNet
```

### pretrain checkpoint download
[ViT-B SAM model](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth)
Expand Down

0 comments on commit 02328a9

Please sign in to comment.