学习笔记参考苹果开源Depth Pro,0.3秒即可从2D图像生成高分辨率深度图 - 映维网资讯
实况演示亲身体验页面
相关论文DEPTH PRO : SHARP MONOCULAR METRIC DEPTH IN LESS THAN A SECOND
苹果已经将Depth Pro开源,具体请访问GitHub
一、开始
1.设置虚拟环境
conda create -n depth-pro -y python=3.9
conda activate depth-pro
pip install -e ***
指令“pip install -e ***”中“***”需要根据实际写入。
1)使用本地路径:如果你的包在本地,你需要提供包的根目录的路径,此路径下需要包括‘setup.py’或者‘pyproject.toml’。例如,你的包位于/path/to/your/package
,应该使用指令
pip install -e /path/to/your/package
2)使用VCS URL:如果你的包托管在如Git的版本控制系统上,需要提供仓库的URL。例如:
pip install -e git+https://github.com/user/repo.git#egg=PackageName
这里#egg=PackageName是可选的,用于指定包的名称。
2.要下载预训练的检查点,请按照以下代码片段进行操作:
source get_pretrained_models.sh # Files will be downloaded to `checkpoints` directory.
二、从命令行运行
一个辅助脚本来直接在单个图像上运行模型:
# Run prediction on a single image:
depth-pro-run -i ./data/example.jpg
# Run `depth-pro-run -h` for available options.
三、运行结果
输入命令行运行,当进度条到达100%后会跳出结果窗口,如图所示。