EgoHackZero commited on
Commit
361cd34
·
1 Parent(s): afae8e5

upgrade models

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,13 +7,13 @@ from PIL import Image
7
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
8
 
9
  # Загрузка модели
10
- midas = torch.hub.load("intel-isl/MiDaS", "MiDaS_small")
11
  midas.to(device)
12
  midas.eval()
13
 
14
  # Загрузка трансформаций
15
  midas_transforms = torch.hub.load("intel-isl/MiDaS", "transforms")
16
- transform = midas_transforms.small_transform
17
 
18
  def predict_depth(image):
19
  # ======= 1. Преобразование в OpenCV формат =======
 
7
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
8
 
9
  # Загрузка модели
10
+ midas = torch.hub.load("intel-isl/MiDaS", "DPT_Large")
11
  midas.to(device)
12
  midas.eval()
13
 
14
  # Загрузка трансформаций
15
  midas_transforms = torch.hub.load("intel-isl/MiDaS", "transforms")
16
+ transform = midas_transforms.dpt_transform
17
 
18
  def predict_depth(image):
19
  # ======= 1. Преобразование в OpenCV формат =======