PytorchNetHub
PytorchNetHub copied to clipboard
项目注释+论文复现+算法竞赛+Pytorch实践
trafficstars
目的
-
论文复现
-
算法竞赛
-
最小实践
-
项目注释
paper得来终觉浅,绝知此事要coding。
培养阅读源码的能力
工作
轻量级图像识别框架Classification
支持任务: 1. 图像分类 2. 度量学习/特征对比
轻量级、模块化、高扩展、分布式、自动剪枝
工具库bobotools
Pytorch工具torch_tools
图像工具img_tools
文本工具txt_tools
列表工具list_tools
Pytorch最小实践
自动混合精度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
汇总
2022
| 官方库 | 备注 | 更新 |
|---|---|---|
| RepVGG(CVPR 2021) | 让VGG再次伟大 解读 | 2022.7 |
| G-Ghost(IJCV 2022) | 探索各级别的特征冗余 解读 | 2022.7 |
| 自研 | 备注 | 更新 |
|---|---|---|
| bobotools |
工具库 | 2022.2 |
| Classification |
图像识别框架 | 2022.2 |
| Pytorch最小实践 |
Pytorch最小实践 | 2022.6 |
| BossVision | 简单、模块化、高扩展的分布式训练框架 | 2022.7 |
2021
| 官方库 | 项目注释 | 备注 | 更新 |
|---|---|---|---|
| SST(ECCV 2020) | 注释 | 浅层人脸学习 | 2021.2 |
| RepDistiller | 注释 | 知识蒸馏算法合集 | 2021.2 |
| 自研 | 备注 | 更新 |
|---|---|---|
| JittorDogsClass |
“计图”算法挑战赛-狗细分类 4/430 | 2021.4 |
| Yolov5-Face |
人脸检测-支持全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) |
新型激活函数-复现 | 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 |
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) | 注释 | 特征金字塔 |
| 自研 | 备注 |
|---|---|
| XueLangTianchi |
雪浪制造AI挑战赛—视觉计算辅助良品检测 -32/2403 |
| S4ND |
单次单尺度肺结节检测(MICCAI 2018)复现 |
| 3D SKconv |
注意力机制SE模块、SK模块的3D实现 |
注:猫狗分类、风格迁移、生成对抗等更多内容请访问pytorch-book