From af5e3a6699fce0e6aee168e60bf3380c1347a783 Mon Sep 17 00:00:00 2001 From: Andrei Luzan <3865242+JIy3AHKO@users.noreply.github.com> Date: Thu, 1 Feb 2024 14:38:25 +0200 Subject: [PATCH] Update README.md Add missing import to example code snippet --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index dfe3b599..78956021 100644 --- a/README.md +++ b/README.md @@ -151,6 +151,7 @@ from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet import cv2 import torch +from torchvision.transforms import Compose encoder = 'vits' # can also be 'vitb' or 'vitl' depth_anything = DepthAnything.from_pretrained('LiheYoung/depth_anything_{:}14'.format(encoder)).eval()