pustar / pytorchnethub

项目注释+论文复现+算法竞赛+Pytorch指北

目的

• 论文复现

• 算法竞赛

• 项目注释

  即开即用推荐官方库,阅读源码推荐注释

• Pytorch指北

接下来工作

目标检测

• Faster rcnn
• YOLO v1
• YOLO v3
• YOLO v4
• SSD
• CAM
• S4ND

语义分割

• U-Net

主干网络

• FPN
• SKNet
• HS-ResNet
• AFF-ResNet 并入官方库
• TargetDrop

激活函数

• FunnelAct
• DynamicReLU

人脸识别

• Semi-Siamese-Training
• BroadFace

人脸损失

• AMSoftmax
• ArcFace
• Circle-Loss
• DiscFace
• NPCFace

知识蒸馏

• RepDistiller

算法竞赛

• 雪浪制造AI挑战赛 排名: 32/2403
• “计图”人工智能算法挑战赛-狗细分类 排名: 4/430

Pytorch指北

• 自动混合精度AMP

• 分布式数据并行DDP

• 模型统计

  统计模型 计算量|参数量|显存占用|耗时(CPU/GPU)

• 移动端部署

  Pytorch -> ONNX -> NCNN or MNN

• 服务端部署

  Pytorch -> TensorRT

  Pytorch -> ONNX -> TensorRT

汇总

2021

模型	官方库	注释	实现	来源	备注	更新
SST	Link	注释		ECCV 2020	浅层人脸学习	2021.2
RepDistiller	Link	注释			知识蒸馏库	2021.2
JittorDogsClass			实现		狗细分类竞赛	2021.4

2020

模型	官方库	注释	实现	来源	备注	更新
FunnelRelu	MegEngine		复现	ECCV 2020	较PRelu等更有效	2020.7
DynamicReLU	Link	注释		ECCV 2020	动态激活函数	2020.9
AMSoftmax	Link	注释			乘法角间隔	2020.9
ArcFace	Link	注释		CVPR 2019	加法角间隔	2020.9
CircleLoss	Link	注释		CVPR 2020	加权角间隔	2020.9
AMP			实现		自动混合精度	2020.10
BroadFace			复现	ECCV 2020	队列更新	2020.10
TargetDrop			复现		注意力Drop	2020.10
HS-ResNet			复现		改进ResNet	2020.11
AFF-ResNet	MXNet		复现	WACV 2021	统一特征融合	2020.11
DDP			实现		分布式数据并行	2020.11
DiscFace			复现	ACCV 2020	最小差异学习	2020.12
NPCFace			复现		正负联合监督	2020.12

2017-2019

模型	官方库	注释	实现	来源
Fatser Rcnn	Link	注释		NIPS 2015
YOLO v1	Link	注释		CVPR 2016
YOLO v3	Link	注释	重构
YOLO v4	Link	注释
SSD	Link	注释	重构	ECCV 2016
CAM	Link	注释		CVPR 2016
S4ND			复现	MICCAI 2018
U-Net	Link	注释		MICCAI 2015
SKNet	Link		实现	CVPR 2019
FPN	Link	注释		CVPR 2017
XueLangTianchi			实现	雪浪制造AI挑战赛

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