Deep Learning

深度学习模型基于pytorch的实现

个人笔记

图像相关任务

神经网络-CNN基础

神经网络-CNN系列

目标检测-RCNN系列

目标检测-YOLO系列

生成模型-GAN基础

生成模型-GAN系列

生成模型-Diffusion Model基础

文本相关任务

神经网络-RNN系列

神经网络-AE&VAE基础

神经网络-seq2seq

神经网络-transformer

预训练-BERT

预训练-GPT系列

任务流程相关

数学工具-常用分类任务评估指标

数学工具-常用回归任务评估指标

数学工具-常用聚类任务评估指标

数学工具-常用目标检测任务评估指标

目录结构

.
├── data_parse  # 数据解析类
├── examples    # 任务使用类
├── metrics     # 模型评估类
├── models      # 模型实现类
├── processor   # 流程处理类
└── utils       # 通用工具类

运行环境

所有脚本均在python==3.8, pytorch==1.11, torchvision==0.12 环境下运行通过