NightRaven109 commited on
Commit
d5707b7
·
verified ·
1 Parent(s): 6b21338

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -11,7 +11,8 @@ from depth_anything_v2.dpt import DepthAnythingV2
11
  model_configs = {
12
  'vits': {'encoder': 'vits', 'features': 64, 'out_channels': [48, 96, 192, 384]},
13
  'vitb': {'encoder': 'vitb', 'features': 128, 'out_channels': [96, 192, 384, 768]},
14
- 'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]}
 
15
  }
16
 
17
  class NormalMapSimple:
@@ -179,7 +180,7 @@ def process_normal_map(image):
179
 
180
  # Download and initialize model
181
  def initialize_model():
182
- encoder = 'vitl'
183
  max_depth = 1
184
 
185
  model = DepthAnythingV2(**{**model_configs[encoder], 'max_depth': max_depth})
 
11
  model_configs = {
12
  'vits': {'encoder': 'vits', 'features': 64, 'out_channels': [48, 96, 192, 384]},
13
  'vitb': {'encoder': 'vitb', 'features': 128, 'out_channels': [96, 192, 384, 768]},
14
+ 'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]},
15
+ 'vitg': {'encoder': 'vitg', 'features': 384, 'out_channels': [1536, 1536, 1536, 1536]}
16
  }
17
 
18
  class NormalMapSimple:
 
180
 
181
  # Download and initialize model
182
  def initialize_model():
183
+ encoder = 'vitg'
184
  max_depth = 1
185
 
186
  model = DepthAnythingV2(**{**model_configs[encoder], 'max_depth': max_depth})