github深度学习项目复现教程

如何找感兴趣的项目?

使用github或papers with code搜索关键词"deep learning"等,最受欢迎的是stars数最多的

查看readme

是否清晰地描述了项目目标、使用的技术、安装步骤和运行方法

是否包含依赖项、所需数据集和训练模型等信息

1、准备环境

如果是租用gpu服务器,可以直接选好拥有对应环境的服务器,就不用再新建虚拟环境占用空间了(服务器ssh登录vscode时,如果连接不上,可能是因为配置文件里已经有了相同服务器的记录,要把它注释掉,再连接)

(1)克隆项目到本地

终端输入:git clone 项目url(克隆到当前文件夹下)

或者下载zip解压缩,再用vscode打开

推荐使用git clone,下载的是完整的仓库,可以自动检查更新和完整性(如部分文件已经下载,重新克隆时会判断是否需要更新这些文件),而zip只是一个静态文件包

(2)创建并激活环境

conda create -n unet python=3.9

conda activate unet

复制代码
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

(pytorch官网找到对应的torch版本的下载命令,服务器拥有的cuda是12.2,所以下载命令使用的版本不能超过12.2)

(3)安装依赖项

requirments.txt或environment.yml

pip install -r requirments.txt (-r表示read,读取文件)

使用envirment.yml时,

在上一步创建环境的时候这样写conda env create -n myenv --file environment.yml

如果没有依赖文件,就自己去要运行的文件(dataset.pymain.py)里找需要什么依赖,再一个个安装,或者根据它的报错提示来补充安装

2、准备数据

下载数据集到指定文件夹

检查数据集格式,是否需要预处理或格式转换

3、配置和运行项目

(1)仔细阅读配置说明,修改配置文件(config.json或.yaml)

(2)运行项目

比如:pythonn train.py(运行脚本)

可能会报错,需要解决潜在问题

比如这个例子,我们需要把dataset.py里的数据集路径更改成自己的(注意可能有多次出现哦,就要改动很多处)

然后是输入命令python main.py,--action这些是参数,在main.py里可以看到;注意如果不指定--dataset,就会使用default数据集,这里default是liver数据集,如果liver数据集不存在,就会出现ZeroDivisionError: division by zero这样的错误。所以必须指定好数据集。

还要注意此时要进入mian.py所在的目录下,才能找到并执行这个文件

4、检查输出并评估

查看输出(训练日志、生成的模型或可视化结果)

进行评估(有的项目会提供评估脚本或方法)

比较结果(和文档中的参考结果作比较,看是否一致)

相关推荐
李小庆11 小时前
Sowork AI Agent 编程助手教程 :第一章 Python环境搭建与Sowork项目克隆学习目标
github
OpenTiny社区1 天前
🎨 看完 GenUI SDK 源码我悟了!
前端·vue.js·github
千寻girling1 天前
一份不可多得的《微服务》教程
后端·面试·github
霜落长河1 天前
用Gemini提升React代码调试效率的教程
github
英勇无比的消炎药2 天前
TinyRobot 源码深度分析:OpenTiny 的 AI 对话组件库
前端·vue.js·github
逛逛GitHub2 天前
慢慢吃掉你的 Claude Code,在终端里养一只黑洞。
github
jump_jump2 天前
为了重玩金庸群侠传,我研究了一下 Ruffle 怎么复活 Flash
游戏·rust·github
LinXunFeng3 天前
Obsidian - 使用 Share Note 分享笔记并自部署
前端·笔记·github
DayDaydream3 天前
7 天涨了 8000+ Star,Agent Reach 想给 AI 装上互联网眼睛
github
天衍四九4 天前
Git从0到实战(四):冲突解决与版本回退 —— 别怕,出错了也能救
github