Update README.md
Browse files
README.md
CHANGED
@@ -8,7 +8,7 @@ tags:
|
|
8 |
library_name: DanishFungi
|
9 |
license: cc-by-nc-4.0
|
10 |
---
|
11 |
-
# Model card for BVRA/resnet34.
|
12 |
|
13 |
## Model Details
|
14 |
- **Model Type:** Danish Fungi Classification
|
@@ -27,7 +27,7 @@ import torch
|
|
27 |
import torchvision.transforms as T
|
28 |
from PIL import Image
|
29 |
from urllib.request import urlopen
|
30 |
-
model = timm.create_model("hf-hub:BVRA/resnet34.
|
31 |
model = model.eval()
|
32 |
train_transforms = T.Compose([T.Resize((299, 299)),
|
33 |
T.ToTensor(),
|
|
|
8 |
library_name: DanishFungi
|
9 |
license: cc-by-nc-4.0
|
10 |
---
|
11 |
+
# Model card for BVRA/resnet34.in1k_ft_df20m_299
|
12 |
|
13 |
## Model Details
|
14 |
- **Model Type:** Danish Fungi Classification
|
|
|
27 |
import torchvision.transforms as T
|
28 |
from PIL import Image
|
29 |
from urllib.request import urlopen
|
30 |
+
model = timm.create_model("hf-hub:BVRA/resnet34.in1k_ft_df20m_299", pretrained=True)
|
31 |
model = model.eval()
|
32 |
train_transforms = T.Compose([T.Resize((299, 299)),
|
33 |
T.ToTensor(),
|