ENSEEIHT/projet-long
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Update README

Browse source 
Former-commit-id: 8ebf8f874be976cace2d32365c8dce31fe7d6d5a
This commit is contained in:
milesial 2021-08-19 11:14:36 +02:00 committed by GitHub
parent 1c450fa978
commit 1f1967d271
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
README.md
View file

@ -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).