-
论文复现
-
算法竞赛
-
项目注释
-
Pytorch指北
-
常用工具
持续更新
最小实践：自动混合精度AMP、分布式数据并行DDP、TensorRT加速、移动端NCNN/MNN
模型统计: 计算量|参数量|耗时
目标检测：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
| 自研 | 备注 | 更新 |
|---|---|---|
| ToolsLib | 常用工具库 | 2022.2 |
Classification |
轻量级图像识别框架 | 2022.2 |
| 官方库 | 项目注释 | 备注 | 更新 |
|---|---|---|---|
| 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