buaacyw / GaussianEditor

[CVPR 2024] GaussianEditor: Swift and Controllable 3D Editing with Gaussian Splatting
https://buaacyw.github.io/gaussian-editor/
Other
1.07k stars 54 forks source link

Could you tell me where to download "omnidata_dpt_depth_v2.ckpt"? #15

Closed MnKnight1 closed 9 months ago

MnKnight1 commented 9 months ago

I see that I have to download omnidata_dpt_depth_v2.ckpt in GaussianEditor-master/threestudio/utils/dpt.py Because of the server network, I had to load it locally and could not download it online,could you tell me where to download "omnidata_dpt_depth_v2.ckpt"?Is it same as https://github.com/ashawkey/stable-dreamfusion/blob/5550b91862a3af7842bb04875b7f1211e5095a63/readme.md?plain=1#L68

         if self.mode == "depth":
            path = ".cache/dpt/omnidata_dpt_depth_v2.ckpt"
            self.model = DPTDepthModel(backbone="vitb_rn50_384")
            self.aug = transforms.Compose(
                [
                    transforms.Resize((384, 384)),
                    transforms.Normalize(mean=0.5, std=0.5),
                ]
            )
        elif self.mode == "normal":
            path = ".cache/dpt/omnidata_dpt_normal_v2.ckpt"
            self.model = DPTDepthModel(backbone="vitb_rn50_384", num_channels=3)
            self.aug = transforms.Compose(
                [
                    transforms.Resize((384, 384)),
                ]
            )
zhanghongyong123456 commented 9 months ago

I see that I have to download omnidata_dpt_depth_v2.ckpt in GaussianEditor-master/threestudio/utils/dpt.py Because of the server network, I had to load it locally and could not download it online,could you tell me where to download "omnidata_dpt_depth_v2.ckpt"?Is it same as https://github.com/ashawkey/stable-dreamfusion/blob/5550b91862a3af7842bb04875b7f1211e5095a63/readme.md?plain=1#L68

         if self.mode == "depth":
            path = ".cache/dpt/omnidata_dpt_depth_v2.ckpt"
            self.model = DPTDepthModel(backbone="vitb_rn50_384")
            self.aug = transforms.Compose(
                [
                    transforms.Resize((384, 384)),
                    transforms.Normalize(mean=0.5, std=0.5),
                ]
            )
        elif self.mode == "normal":
            path = ".cache/dpt/omnidata_dpt_normal_v2.ckpt"
            self.model = DPTDepthModel(backbone="vitb_rn50_384", num_channels=3)
            self.aug = transforms.Compose(
                [
                    transforms.Resize((384, 384)),
                ]
            )

you can download https://huggingface.co/camenduru/PeRF/tree/main

heheyas commented 9 months ago

Check #10 for manually downloading the weight for the depth estimator.

Reference

MnKnight1 commented 9 months ago

Thank you for your help!