Skip to content

Commit 338fc23

Browse files
committed
Added imagenet normalization to dinov2small custom model example
1 parent 904dd2d commit 338fc23

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

examples/dinov2small.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,9 @@ def __init__(self):
1313
[
1414
transforms.ToTensor(),
1515
transforms.Resize((224, 224)),
16+
transforms.Normalize(
17+
mean=(0.485, 0.456, 0.406), std=(0.229, 0.224, 0.225)
18+
),
1619
]
1720
)
1821
self.name = "dinov2_vits14"

0 commit comments

Comments
 (0)