mirror of
https://github.com/Laurent2916/REVA-QCAV.git
synced 2024-11-09 15:02:03 +00:00
Quick start without docker
Former-commit-id: c4555b6d1ae960f8d253b1e45c25e2279b64ead5
This commit is contained in:
parent
a445c4bdc4
commit
02fc4ff733
25
README.md
25
README.md
|
@ -9,7 +9,9 @@
|
||||||
|
|
||||||
Customized implementation of the [U-Net](https://arxiv.org/abs/1505.04597) in PyTorch for Kaggle's [Carvana Image Masking Challenge](https://www.kaggle.com/c/carvana-image-masking-challenge) from high definition images.
|
Customized implementation of the [U-Net](https://arxiv.org/abs/1505.04597) in PyTorch for Kaggle's [Carvana Image Masking Challenge](https://www.kaggle.com/c/carvana-image-masking-challenge) from high definition images.
|
||||||
|
|
||||||
- [Quick start using Docker](#quick-start-using-docker)
|
- [Quick start](#quick-start)
|
||||||
|
- [Without Docker](#without-docker)
|
||||||
|
- [With Docker](#with-docker)
|
||||||
- [Description](#description)
|
- [Description](#description)
|
||||||
- [Usage](#usage)
|
- [Usage](#usage)
|
||||||
- [Docker](#docker)
|
- [Docker](#docker)
|
||||||
|
@ -19,7 +21,26 @@ Customized implementation of the [U-Net](https://arxiv.org/abs/1505.04597) in Py
|
||||||
- [Pretrained model](#pretrained-model)
|
- [Pretrained model](#pretrained-model)
|
||||||
- [Data](#data)
|
- [Data](#data)
|
||||||
|
|
||||||
## Quick start using Docker
|
## Quick start
|
||||||
|
|
||||||
|
### Without Docker
|
||||||
|
|
||||||
|
1. [Install CUDA](https://developer.nvidia.com/cuda-downloads)
|
||||||
|
|
||||||
|
2. [Install PyTorch](https://pytorch.org/get-started/locally/)
|
||||||
|
|
||||||
|
3. Install dependencies
|
||||||
|
```bash
|
||||||
|
pip install -r requirements.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
4. Download the data and run training:
|
||||||
|
```bash
|
||||||
|
bash scripts/download_data.sh
|
||||||
|
python train.py --amp
|
||||||
|
```
|
||||||
|
|
||||||
|
### With Docker
|
||||||
|
|
||||||
1. [Install Docker 19.03 or later:](https://docs.docker.com/get-docker/)
|
1. [Install Docker 19.03 or later:](https://docs.docker.com/get-docker/)
|
||||||
```bash
|
```bash
|
||||||
|
|
Loading…
Reference in a new issue