(深度估计学习)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 git@github.com: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.运行结果

相关推荐
艾莉丝努力练剑10 分钟前
【LeetCode&数据结构】单链表的应用——反转链表问题、链表的中间节点问题详解
c语言·开发语言·数据结构·学习·算法·leetcode·链表
人生游戏牛马NPC1号2 小时前
学习 Flutter (三):玩安卓项目实战 - 上
android·学习·flutter
没有羊的王K5 小时前
SSM框架学习——day1
java·学习
林林要一直努力7 小时前
AOSP Settings模块问题初窥
android·学习·bug·android studio
余大侠在劈柴9 小时前
pdf.js 开发指南:在 Web 项目中集成 PDF 预览功能
前端·javascript·学习·pdf
有谁看见我的剑了?11 小时前
iperf3 网络带宽测试工具学习
学习·测试工具
老神在在00111 小时前
SpringMVC2
java·前端·学习·spring·java-ee
老神在在00111 小时前
SpringMVC3
java·前端·学习·spring·java-ee
Pocker_Spades_A15 小时前
TextIn:文档全能助手,让学习效率飙升的良心软件~
学习·textln
我.佛.糍.粑16 小时前
Shusen Wang推荐系统学习 --召回 矩阵补充 双塔模型
人工智能·学习·机器学习·矩阵·推荐算法