(深度估计学习)Win11复现DepthFM

目录

论文链接:https://depthfm.github.io/

讲解链接:https://www.php.cn/faq/734404.html

1. 系统配置

本人系统:Win11 CUDA12.2 python3.11.5

这里附上几个CUDA安装链接:

  1. 安装CUDA和CudaNN
  2. CUDA Toolkit 12.2 Update 2 Downloads
  3. cuDNN Archive,这里我选择的:Download cuDNN v8.9.7 (December 5th, 2023), for CUDA 12.x
  4. windows下切换不同版本cuda

2. 拉取代码,配置环境

首先,克隆 github 仓库

shell 复制代码
git clone [email protected]:CompVis/depth-fm.git
cd depth-fm

下载权重

shell 复制代码
Invoke-WebRequest -Uri "https://ommer-lab.com/files/depthfm/depthfm-v1.ckpt" -OutFile "checkpoints/depthfm-v1.ckpt"                          

创建 conda 环境。这里我之前的环境是cuda11.8就失败了,然后换到了cuda12.2

python 复制代码
conda env create -f environment.yml

3.开始深度预测

python 复制代码
python inference.py \
   --num_steps 2 \
   --ensemble_size 4 \
   --img assets/dog.png \
   --ckpt checkpoints/depthfm-v1.ckpt

但是此时报错了:
OSError: runwayml/stable-diffusion-v1-5 does not appear to have a file named config.json.

在项目的issue中找到了解决方法:https://github.com/CompVis/depth-fm/issues/16

一句命令解决huggingface.co无法下载模型问题

在windows里面直接设置环境变量

接下来重新运行python inference.py

4.运行结果

相关推荐
云上艺旅5 小时前
K8S学习之基础七十四:部署在线书店bookinfo
学习·云原生·容器·kubernetes
你觉得2055 小时前
哈尔滨工业大学DeepSeek公开课:探索大模型原理、技术与应用从GPT到DeepSeek|附视频与讲义下载方法
大数据·人工智能·python·gpt·学习·机器学习·aigc
A旧城以西6 小时前
数据结构(JAVA)单向,双向链表
java·开发语言·数据结构·学习·链表·intellij-idea·idea
无所谓จุ๊บ7 小时前
VTK知识学习(50)- 交互与Widget(一)
学习·vtk
FAREWELL000757 小时前
C#核心学习(七)面向对象--封装(6)C#中的拓展方法与运算符重载: 让代码更“聪明”的魔法
学习·c#·面向对象·运算符重载·oop·拓展方法
吴梓穆7 小时前
UE5学习笔记 FPS游戏制作38 继承标准UI
笔记·学习·ue5
Three~stone7 小时前
MySQL学习集--DDL
数据库·sql·学习
齐尹秦8 小时前
HTML 音频(Audio)学习笔记
学习
瞌睡不来8 小时前
(学习总结32)Linux 基础 IO
linux·学习·io
Moonnnn.8 小时前
运算放大器(四)滤波电路(滤波器)
笔记·学习·硬件工程