verkaDerkaDerk commited on
Commit
760a850
1 Parent(s): 34693bd

change to ZoeD_K after ZoeD_N mysteriously stopped working

Browse files
Files changed (1) hide show
  1. 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', "ZoeD_N", 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)
 
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)