Spaces:
Runtime error
Runtime error
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)
|