Update README.md
Browse files
README.md
CHANGED
|
@@ -43,6 +43,7 @@ Trained by [Tissue Image Analytics (TIA) Centre](https://warwick.ac.uk/fac/cross
|
|
| 43 |
```python
|
| 44 |
from urllib.request import urlopen
|
| 45 |
from PIL import Image
|
|
|
|
| 46 |
import timm
|
| 47 |
|
| 48 |
# get example histology image
|
|
@@ -74,6 +75,7 @@ output = model(data) # output is a (batch_size, num_features) shaped tensor
|
|
| 74 |
```python
|
| 75 |
from urllib.request import urlopen
|
| 76 |
from PIL import Image
|
|
|
|
| 77 |
import timm
|
| 78 |
|
| 79 |
# get example histology image
|
|
|
|
| 43 |
```python
|
| 44 |
from urllib.request import urlopen
|
| 45 |
from PIL import Image
|
| 46 |
+
import torch.nn as nn
|
| 47 |
import timm
|
| 48 |
|
| 49 |
# get example histology image
|
|
|
|
| 75 |
```python
|
| 76 |
from urllib.request import urlopen
|
| 77 |
from PIL import Image
|
| 78 |
+
import torch.nn as nn
|
| 79 |
import timm
|
| 80 |
|
| 81 |
# get example histology image
|