-
论文复现
-
算法竞赛
-
源码注释
paper得来终觉浅，绝知此事要coding。
支持任务： 1. 图像分类 2. 度量学习/特征对比
轻量级、模块化、高扩展、分布式、自动剪枝
Pytorch工具torch_tools、图像工具img_tools、文本工具txt_tools、列表工具list_tools
自动混合精度AMP
分布式数据并行DDP
NCNN/MNN部署
TensorRT部署
目标检测：Faster rcnn、YOLO v1、YOLO v3、YOLO v4、SSD、CAM、S4ND
语义分割：U-Net
主干网络：FPN、SKNet、HS-ResNet、AFF-ResNet(并入官方库)、TargetDrop
激活函数：FunnelAct、DynamicReLU
知识蒸馏：RepDistiller
人脸检测：Yolov5-Face 主要贡献者
人脸比对
分类器：AMSoftmax|ArcFace|Circle-Loss|DiscFace|NPCFace
训练策略：Semi-Siamese-Training|BroadFace
雪浪制造AI挑战赛 排名: 32/2403
"计图"人工智能算法挑战赛-狗细分类 排名: 4/430
ACCV2022国际细粒度图像分析挑战赛-网络监督的细粒度识别 排名: 8/133
| 自研 | 备注 | 更新 |
|---|---|---|
| 多模态大模型专栏 | 多模态VL系列之高质量数据（一） 多模态VL系列之模型架构（二） 多模态VL系列之训练策略（三） |
2024.4 |
| 24卡+公开数据+4B 可以有多强？ | 勇闯OpenCompass榜单 | 2024.8 |
| 自研 | 备注 | 更新 |
|---|---|---|
| LearnColossalAI | ColossalAI教程 & 示例注释 & 学习笔记 （大模型高效训练） | 2023.3 |
| OpenMM | OpenMM系列最佳食用手册 | 2023.3 |
LearnDeepSpeed |
DeepSpeed教程 & 示例注释 & 学习笔记 （大模型高效训练） | 2023.8 |
| MiniGPT-4-DeepSpeed | MiniGPT-4基于DeepSpeed加速➕ 扩充模型规模 ➕ 实验分析 | 2023.9 |
| 官方库 | 更新 | |
|---|---|---|
| RepVGG(CVPR 2021) | VGG再次伟大 解读+代码 | 2022.7 |
| G-Ghost(IJCV 2022) | 探索各级别的特征冗余 解读+代码 | 2022.7 |
| BBN(CVPR2020) | 双边分支网络,以解决长尾分布问题 解读+代码 | 2022.11 |
| Vision Transformer(ICLR 2021) | Transformer首次应用到视觉领域 解读+代码 | 2022.12 |
| MAE(CVPR2021) | 自监督学习 解读+代码 | 2022.12 |
| 自研 | 备注 | 更新 |
|---|---|---|
bobotools |
工具库 | 2022.2 |
Classification |
图像识别框架 | 2022.2 |
Pytorch最小实践 |
Pytorch最小实践 | 2022.6 |
| BossVision | 简单、模块化、高扩展的分布式训练框架 | 2022.7 |
| CUDA-Python | Nvidia CUDA加速计算基础课程 | 2022.9 |
| DataHub | 公开数据集汇总 | 2022.11 |
| 官方库 | 项目注释 | 备注 | 更新 |
|---|---|---|---|
| SST(ECCV 2020) | 注释 | 浅层人脸学习 | 2021.2 |
| RepDistiller | 注释 | 知识蒸馏算法合集 | 2021.2 |
| 自研 | 备注 | 更新 |
|---|---|---|
JittorDogsClass |
“计图”算法挑战赛-狗细分类 4/430 | 2021.4 |
Yolov5-Face |
人脸检测-支持纯TensorRT加速，成为主要贡献者 | 2021.12 |
| 官方库 | 项目注释 | 备注 | 更新 |
|---|---|---|---|
| DynamicReLU(ECCV 2020) | 注释 | 动态激活函数 | 2020.9 |
| AMSoftmax | 注释 | 乘法角间隔 | 2020.9 |
| ArcFace(CVPR 2019) | 注释 | 加法角间隔 | 2020.9 |
| CircleLoss(CVPR 2020) | 注释 | 加权角间隔 | 2020.9 |
| 自研 | 备注 | 更新 |
|---|---|---|
FunnelRelu(ECCV 2020) |
新型激活函数-复现 | 2020.7 |
| AMP |
自动混合精度-示例 | 2020.10 |
BroadFace(ECCV 2020) |
人脸对比队列更新策略-复现 | 2020.10 |
TargetDrop |
注意力机制Dropout-复现 | 2020.10 |
HS-ResNet |
ResNet改进版-复现 | 2020.11 |
AFF-ResNet(WACV 2021) |
特征融合的统一方式- 复现，并入官方库 | 2020.11 |
| DDP |
分布式数据并行-示例 | 2020.11 |
DiscFace(ACCV 2020) |
最小差异学习-复现 | 2020.12 |
| NPCFace |
正负联合监督-复现 | 2020.12 |
| 官方库 | 项目注释 | 备注 |
|---|---|---|
| Fatser Rcnn(NIPS 2015) | 注释 | 目标检测 |
| YOLO v1(CVPR 2016) | 注释 | 目标检测 |
| YOLO v3(ECCV 2016) | 注释 重构 | 目标检测 |
| YOLO v4 | 注释 | 目标检测 |
| SSD | 注释 重构 | 目标检测 |
| CAM(CVPR 2016) | 注释 | 特征可视化 |
| U-Net(MICCAI 2015) | 注释 | 医学影像语义分割 |
| FPN(CVPR 2017) | 注释 | 特征金字塔 |
| 自研 | 备注 |
|---|---|
XueLangTianchi |
雪浪制造AI挑战赛—视觉计算辅助良品检测 -32/2403 |
S4ND  |
单次单尺度肺结节检测（MICCAI 2018）复现 |
3D SKconv  |
注意力机制SE模块、SK模块的3D实现 |
注：猫狗分类、风格迁移、生成对抗等更多内容请访问pytorch-book