Hugging Face's logo

Spaces:
blanchon
/
Metric3D
Paused

App Files Community
JUGGHM commited on
Commit
48a1135
·
1 Parent(s): 18e9512

Update mono/configs/_base_/models/encoder_decoder/dino_vit_large_reg.dpt_raft.py

Browse files
Files changed (1) hide show
  1. mono/configs/_base_/models/encoder_decoder/dino_vit_large_reg.dpt_raft.py +1 -1
mono/configs/_base_/models/encoder_decoder/dino_vit_large_reg.dpt_raft.py CHANGED
@@ -3,7 +3,7 @@ _base_ = ['../backbones/dino_vit_large_reg.py']
3
  model = dict(
4
  type='DensePredModel',
5
  decode_head=dict(
6
- type='RAFTDepthDPT',
7
  in_channels=[1024, 1024, 1024, 1024],
8
  use_cls_token=True,
9
  feature_channels = [256, 512, 1024, 1024], # [2/7, 1/7, 1/14, 1/14]
 
3
  model = dict(
4
  type='DensePredModel',
5
  decode_head=dict(
6
+ type='RAFTDepthNormalDPT5',
7
  in_channels=[1024, 1024, 1024, 1024],
8
  use_cls_token=True,
9
  feature_channels = [256, 512, 1024, 1024], # [2/7, 1/7, 1/14, 1/14]