Commit ·
101eab1
1
Parent(s): 5366461
load model externally
Browse files- .DS_Store +0 -0
- checkpoints/.DS_Store +0 -0
- depth_anything/dpt.py +1 -1
.DS_Store
CHANGED
|
Binary files a/.DS_Store and b/.DS_Store differ
|
|
|
checkpoints/.DS_Store
CHANGED
|
Binary files a/checkpoints/.DS_Store and b/checkpoints/.DS_Store differ
|
|
|
depth_anything/dpt.py
CHANGED
|
@@ -137,7 +137,7 @@ class DPTHead(nn.Module):
|
|
| 137 |
|
| 138 |
|
| 139 |
class DPT_DINOv2(nn.Module):
|
| 140 |
-
def __init__(self, encoder='vitl', features=256, out_channels=[256, 512, 1024, 1024], use_bn=False, use_clstoken=False, localhub=
|
| 141 |
super(DPT_DINOv2, self).__init__()
|
| 142 |
|
| 143 |
assert encoder in ['vits', 'vitb', 'vitl']
|
|
|
|
| 137 |
|
| 138 |
|
| 139 |
class DPT_DINOv2(nn.Module):
|
| 140 |
+
def __init__(self, encoder='vitl', features=256, out_channels=[256, 512, 1024, 1024], use_bn=False, use_clstoken=False, localhub=False):
|
| 141 |
super(DPT_DINOv2, self).__init__()
|
| 142 |
|
| 143 |
assert encoder in ['vits', 'vitb', 'vitl']
|