mirror of
https://github.com/Laurent2916/REVA-QCAV.git
synced 2024-11-09 15:02:03 +00:00
Update README
Former-commit-id: 8ebf8f874be976cace2d32365c8dce31fe7d6d5a
This commit is contained in:
parent
1c450fa978
commit
1f1967d271
|
@ -138,12 +138,12 @@ When launching a training, a link will be printed in the console. Click on it to
|
||||||
|
|
||||||
|
|
||||||
## Pretrained model
|
## Pretrained model
|
||||||
A [pretrained model](https://github.com/milesial/Pytorch-UNet/releases/tag/v1.0) is available for the Carvana dataset. It can also be loaded from torch.hub:
|
A [pretrained model](https://github.com/milesial/Pytorch-UNet/releases/tag/v2.0) is available for the Carvana dataset. It can also be loaded from torch.hub:
|
||||||
|
|
||||||
```python
|
```python
|
||||||
net = torch.hub.load('milesial/Pytorch-UNet', 'unet_carvana')
|
net = torch.hub.load('milesial/Pytorch-UNet', 'unet_carvana')
|
||||||
```
|
```
|
||||||
The training was done with a 100% scale and bilinear upsampling.
|
The training was done with a 50% scale and bilinear upsampling.
|
||||||
|
|
||||||
## Data
|
## Data
|
||||||
The Carvana data is available on the [Kaggle website](https://www.kaggle.com/c/carvana-image-masking-challenge/data).
|
The Carvana data is available on the [Kaggle website](https://www.kaggle.com/c/carvana-image-masking-challenge/data).
|
||||||
|
|
Loading…
Reference in a new issue