【Ultra-Fast-Lane-Detection-v2代码实现】

Ultra-Fast-Lane-Detection-v2代码实现

---

---

前言

第一次写，别骂我

---

一、背景介绍

做毕设中，需要先跑几个模型起来
paper：https://arxiv.org/pdf/2206.07389.pdf
代码链接：https://github.com/cfzd/Ultra-Fast-Lane-Detection-v2
论文解读（我的解读，正常人别看）：暂时还没有

二、环境搭建

1.克隆项目

下载代码（要是这个代码不太行就直接下载哈）：

git clone https://github.com/cfzd/Ultra-Fast-Lane-Detection-V2
cd Ultra-Fast-Lane-Detection-V2

2.创建 conda 虚拟环境并将其激活

conda create -n lane-det python=3.7 -y
conda activate lane-det

3.安装依赖项

# If you dont have pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

注意：这里如果这条代码没法直接下载就是cuda和pytorch版本不对应

处理方法：
1. 查看版本：

方式一：conda list 可以看到pytorch版本和cuda的版本
方式二：在python代码中验证查看

print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch