-
论文复现
-
算法竞赛
-
源码注释
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
自研 | 备注 |
---|---|
bobotools |
工具库(推荐Chat-GPT) |
OpenMM | OpenMM系列最佳食用手册 |
DataHub | 公开数据集Hub |
LearnDeepSpeed | DeepSpeed教程 & 示例注释 & 学习笔记 (大模型高效训练) |
自研 | 备注 | 更新 |
---|---|---|
VLM | 多模态大模型 数据&训练&评估&认知 | 2024.7 |
自研 | 备注 | 更新 |
---|---|---|
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