verkaDerkaDerk
commited on
Commit
•
760a850
1
Parent(s):
34693bd
change to ZoeD_K after ZoeD_N mysteriously stopped working
Browse files- MediaMesh.py +1 -1
MediaMesh.py
CHANGED
@@ -311,7 +311,7 @@ class ZoeDepthSource( DepthSource ):
|
|
311 |
|
312 |
def __init__(self):
|
313 |
super().__init__(ZoeDepthSource.NAME)
|
314 |
-
self.model = torch.hub.load('isl-org/ZoeDepth', "
|
315 |
|
316 |
def mapDepth(self, mediaMesh:'MediaMesh', depthMapping:DepthMapping=None) -> 'DepthSource':
|
317 |
self.depth = 1.-self.model.infer_pil(mediaMesh.image)
|
|
|
311 |
|
312 |
def __init__(self):
|
313 |
super().__init__(ZoeDepthSource.NAME)
|
314 |
+
self.model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_K", pretrained=True).to(DEVICE).eval()
|
315 |
|
316 |
def mapDepth(self, mediaMesh:'MediaMesh', depthMapping:DepthMapping=None) -> 'DepthSource':
|
317 |
self.depth = 1.-self.model.infer_pil(mediaMesh.image)
|