Hugging Face's logo

Spaces:
JUGGHM
/
Metric3D
Running on T4

App Files Community
8
JUGGHM commited on
Commit
5f766a8
•
1 Parent(s): 48a1135

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

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