Skip to content

bobo0810/PytorchNetHub

Repository files navigation

linus

目的

  • 论文复现

  • 算法竞赛

  • 源码注释

    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

时间线

2024

自研 备注 更新
多模态大模型专栏 多模态VL系列之高质量数据(一)
多模态VL系列之模型架构(二)
多模态VL系列之训练策略(三)
2024.4
24卡+公开数据+4B 可以有多强? 勇闯OpenCompass榜单 2024.8

2023

自研 备注 更新
LearnColossalAI ColossalAI教程 & 示例注释 & 学习笔记 (大模型高效训练) 2023.3
OpenMM OpenMM系列最佳食用手册 2023.3
LearnDeepSpeedGithub stars DeepSpeed教程 & 示例注释 & 学习笔记 (大模型高效训练) 2023.8
MiniGPT-4-DeepSpeed MiniGPT-4基于DeepSpeed加速➕ 扩充模型规模 ➕ 实验分析 2023.9

2022

官方库 更新
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
自研 备注 更新
bobotoolsGithub stars 工具库 2022.2
ClassificationGithub stars 图像识别框架 2022.2
Pytorch最小实践Github stars Pytorch最小实践 2022.6
BossVision 简单、模块化、高扩展的分布式训练框架 2022.7
CUDA-Python Nvidia CUDA加速计算基础课程 2022.9
DataHub 公开数据集汇总 2022.11

2021

官方库 项目注释 备注 更新
SST(ECCV 2020) 注释 浅层人脸学习 2021.2
RepDistiller 注释 知识蒸馏算法合集 2021.2
自研 备注 更新
JittorDogsClassGithub stars “计图”算法挑战赛-狗细分类 4/430 2021.4
Yolov5-FaceGithub stars 人脸检测-支持纯TensorRT加速,成为主要贡献者 2021.12

2020

官方库 项目注释 备注 更新
DynamicReLU(ECCV 2020) 注释 动态激活函数 2020.9
AMSoftmax 注释 乘法角间隔 2020.9
ArcFace(CVPR 2019) 注释 加法角间隔 2020.9
CircleLoss(CVPR 2020) 注释 加权角间隔 2020.9
自研 备注 更新
FunnelRelu(ECCV 2020)Github stars 新型激活函数-复现 2020.7
AMP Github stars 自动混合精度-示例 2020.10
BroadFace(ECCV 2020)Github stars 人脸对比队列更新策略-复现 2020.10
TargetDropGithub stars 注意力机制Dropout-复现 2020.10
HS-ResNetGithub stars ResNet改进版-复现 2020.11
AFF-ResNet(WACV 2021)Github stars 特征融合的统一方式- 复现,并入官方库 2020.11
DDP Github stars 分布式数据并行-示例 2020.11
DiscFace(ACCV 2020)Github stars 最小差异学习-复现 2020.12
NPCFaceGithub stars 正负联合监督-复现 2020.12

2017-2019

官方库 项目注释 备注
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) 注释 特征金字塔
自研 备注
XueLangTianchiGithub stars 雪浪制造AI挑战赛—视觉计算辅助良品检测 -32/2403
S4ND Github stars 单次单尺度肺结节检测(MICCAI 2018)复现
3D SKconv Github stars 注意力机制SE模块、SK模块的3D实现

注:猫狗分类、风格迁移、生成对抗等更多内容请访问pytorch-book

Star History Chart