/PytorchNetHub

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

Primary LanguageJupyter NotebookMIT LicenseMIT

目的

  • 论文复现

  • 算法竞赛

  • 项目注释

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

  • Pytorch指北

接下来工作

目标检测

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

语义分割

  • U-Net

主干网络

激活函数

  • FunnelAct
  • DynamicReLU

人脸检测

人像比对

  • Semi-Siamese-Training
  • BroadFace

人脸损失

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

知识蒸馏

  • RepDistiller

算法竞赛

Pytorch指北

  • 自动混合精度AMP

  • 分布式数据并行DDP

  • 模型统计:计算量|参数量|耗时

  • 移动端部署NCNN/MNN

  • TensorRT最佳实践

  • 数据集格式转换

汇总

2021

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

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