apple
/

DepthPro-mixin

Depth Estimation

model_hub_mixin

pytorch_model_hub_mixin

Model card Files Files and versions Community

amael-apple commited on Oct 5

Commit

8d50045

•

1 Parent(s): 943bb94

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ from depth_pro import create_model_and_transforms, load_rgb
 from depth_pro.depth_pro import (create_backbone_model, load_monodepth_weights,
                                  DepthPro, DepthProEncoder, MultiresConvDecoder)
 import depth_pro
-from torchvision.transforms import Compose, Normalize
 class DepthProWrapper(DepthPro, PyTorchModelHubMixin):

 from depth_pro.depth_pro import (create_backbone_model, load_monodepth_weights,
                                  DepthPro, DepthProEncoder, MultiresConvDecoder)
 import depth_pro
+from torchvision.transforms import Compose, Normalize, ToTensor
 class DepthProWrapper(DepthPro, PyTorchModelHubMixin):