DANNet复现笔记_dan 复现csdn-CSDN博客

本文链接：https://blog.csdn.net/yangming_52/article/details/136244171

环境配置

在环境配置之前，首先要进入主板BIOS关闭安全启动，不然后面装驱动会报错

CUDA安装

NVIDIA驱动安装

ubuntu20.04 安装NVIDIA驱动很容易，只只需要打开系统设置->软件和更新->附加驱动->选择NVIDIA驱动->应用更改。该界面会自动根据电脑上的GPU显示推荐的NVIDIA显卡驱动。

NVIDIA（英伟达）显卡驱动安装完成后，在终端输入nvidia-smi输出如下图所示的结果就表示安装成功了。下图中绿色框内表明，安装的驱动支持的CUDA最高版本为12.0(注意这里需要重新启动电脑)。

安装CUDA

先进一下pytorch官网看一下cuda和pytorch的搭配

Previous PyTorch Versions | PyTorch

所以我选择了CUDA 11.8版本

CUDA官网CUDA Toolkit Archive | NVIDIA Developer

参考文章CSDN中的大佬文章：2023年最新Ubuntu安装pytorch教程

2023年最新Ubuntu安装pytorch教程-CSDN博客

这个地方用到了vim编辑器，安装一下才能用

这里vim编辑器，键盘输入i插入，esc退出编辑，输入：wq保存编辑

跟着教程来安装就好。

安装Anaconda

Unleash AI Innovation and Value | Anaconda

这里不要这样安装

我的做法是将安装包放在/home/ming（主目录）下，右键在终端中打开

bash anaconda3.xxxxx.sh(用Tab键补全)

这里按回车慢一点，不然会漏掉确认

这里看一下安装路径，不要像图片一样安装在root下，如果显示安装在/home/name/anaconda3就是对的。

创建anaconda环境

按照教程来就行

安装pytorch

这边一定要在激活dl环境的条件下安装

就是终端要名称前会带上dl

跟着教程到这就结束了

pycharm安装

跟着大佬的文章安装

在Ubuntu中安装并配置Pycharm教程_ubuntu安装pycharm-CSDN博客

主要是要在配置中选择我们安装了torch的dl环境

至此DANNet所需环境就配置好了

代码复现

GitHub - W-zx-Y/DANNet: (CVPR2021) DANNet: A One-Stage Domain Adaptation Network for Unsupervised Nighttime Semantic Segmentation

下载好解压，用pycharm打开

点进去

测试的话先下载这个包

下载好了解压，最好跟Dannet的代码放在一个文件夹中

根据他的Testing还需要进入这个网站下载一个包

https://www.dropbox.com/s/fmlq806p2wqf311/trained_models.zip?dl=0

Step1: download the [trained models](https://www.dropbox.com/s/fmlq806p2wqf311/trained_models.zip?dl=0) and put it in the root.
Step2: change the data and model paths in configs/test_config.py
Step3: run "python evaluation.py"
Step4: run "python compute_iou.py"