Muyun99's wiki
Muyun99
2021-05-07

资料搜集

# 代码方向可以借鉴的仓库

  • https://github.com/bobo0810/PytorchNetHub
  • https://github.com/Media-Smart/volkscv（带分析和指标）
  • https://mp.weixin.qq.com/s/dXc5fYftceJ8r7MwyAOGOw（CVChain牛逼牛逼）
  • https://github.com/sailist/thexp（自己写的实验框架，可以参考）
  • https://github.com/misads/cv_template（一个自己写的实验框架）
  • https://github.com/eladhoffer/utils.pytorch（一些基于Pytorch的有用的代码片段）
  • https://github.com/CoinCheung/pytorch-loss（各类损失）
  • https://github.com/Megvii-BaseDetection/cvpods（旷视的code base）
  • https://github.com/donnyyou/torchcv
  • https://github.com/PyTorchLightning/metrics（各类metrics）
  • https://bbs.huaweicloud.com/blogs/267402?utm_source=51cto&utm_medium=bbs-ex&utm_campaign=ei&utm_content=content（分类任务评价指标）
  • https://github.com/zhanghang1989/PyTorch-Encoding
  • https://github.com/lxztju/pytorch_classification
  • https://github.com/xuyxu/Ensemble-Pytorch（PyTorch Ensemble的库）
  • https://github.com/xiaoyufenfei/Efficient-Segmentation-Networks
  • https://github.com/CVHuber/Convolution
  • https://github.com/ming71/toolbox
  • https://github.com/BloodAxe/pytorch-toolbelt（一个非常好的竞赛代码template）
  • https://github.com/Alibaba-MIIL/ImageNet21K（ImageNet 21K 的预训练模型库）

# 图像标注

  • https://github.com/lzx1413/labelImgPlus
  • https://github.com/lanbing510/ImageLabel
  • https://github.com/wkentaro/labelme
  • https://github.com/hzylmf/od-annotation
  • https://github.com/rachelcao277/LabelImage
  • https://github.com/abreheret/PixelAnnotationTool

# 向量检索

  • https://github.com/facebookresearch/faiss
  • https://github.com/milvus-io/milvus/
  • https://www.qbitai.com/2021/03/22165.html（Proxima）

# 自监督学习

  • https://github.com/KeremTurgutlu/self_supervised
  • https://github.com/open-mmlab/OpenSelfSup

# 图像处理的库

  • https://github.com/libvips/libvips（速度很快并且不占内存）
  • https://github.com/open-mmlab/mmcv

# 图像分类

  • https://github.com/weiaicunzai/Bag_of_Tricks_for_Image_Classification_with_Convolutional_Neural_Networks

  • https://github.com/weiaicunzai/awesome-image-classification

  • https://github.com/rwightman/pytorch-image-models/（牛逼）

  • https://github.com/PistonY/torch-toolbox

  • https://github.com/spytensor/pytorch_img_classification_for_competition

  • https://github.com/Cadene/pretrained-models.pytorch（牛逼）

  • https://github.com/weiaicunzai/awesome-image-classification

  • https://github.com/PaddlePaddle/PaddleClas#Model_zoo_overview

  • https://github.com/facebookresearch/deit

  • https://github.com/DingXiaoH/RepVGG

  • https://github.com/XingangPan/IBN-Net

  • https://github.com/yifan123/IC-Conv

  • https://github.com/d-li14/involution （RedNet）

  • https://github.com/huawei-noah/CV-Backbones

  • https://github.com/facebookresearch/pycls

  • Attention机制

    • https://github.com/Andrew-Qibin/CoordAttention

  • https://github.com/CyberZHG/torch-multi-head-attention

  • Transformer

    • https://github.com/microsoft/Swin-Transformer
    • https://github.com/zihangJiang/TokenLabeling
    • https://github.com/JDAI-CV/CoTNet

  • 轻量化模型

    • https://github.com/HRNet/Lite-HRNet

  • 细粒度识别

    • https://github.com/FouriYe/CNL-ICIP2020

# 图像分割

  • https://github.com/JunMa11/SegLoss
  • https://github.com/linbo0518/BLSeg
  • https://github.com/qubvel/segmentation_models.pytorch
  • https://github.com/donnyyou/torchcv
  • https://github.com/open-mmlab/mmsegmentation
  • https://github.com/NVIDIA/semantic-segmentation
  • https://github.com/Media-Smart/vedaseg
  • https://github.com/cvlab-yonsei/SFNet
  • https://github.com/Tramac/awesome-semantic-segmentation-pytorch
  • https://github.com/Gsunshine/Enjoy-Hamburger
  • https://github.com/LikeLy-Journey/SegmenTron
  • https://github.com/CSAILVision/semantic-segmentation-pytorch（MIT ADE20K dataset）
  • https://github.com/sacmehta/ESPNet
  • https://github.com/xiaoyufenfei/Efficient-Segmentation-Networks
  • matting 方向
    • https://github.com/PeterL1n/BackgroundMattingV2
  • 实例分割方向
    • https://github.com/Epiphqny/VisTR
    • https://github.com/conradry/max-deeplab
  • 全景分割
    • https://github.com/chensnathan/SpatialFlow
    • https://github.com/w-hc/pcv
  • 自动驾驶
    • https://github.com/voldemortX/pytorch-auto-drive/blob/master/docs/MODEL_ZOO.md（包括semantic segmentation和lane detection）
  • 视频分割
    • https://hkchengrex.github.io/MiVOS/
  • 无监督语义分割
    • https://github.com/wvangansbeke/Unsupervised-Semantic-Segmentation
  • 表面缺陷检测
    • https://github.com/mengcius/Surface-Defect-Detection
  • 半监督语义分割
    • https://github.com/WilhelmT/ClassMix
    • https://github.com/charlesCXK/TorchSemiSeg
    • https://github.com/dvlab-research/Context-Aware-Consistency

# 目标检测

  • https://github.com/hoya012/deep_learning_object_detection
  • https://github.com/hukaixuan19970627/YOLOv5_DOTA_OBB
  • https://github.com/rafaelpadilla/Object-Detection-Metrics
  • 实时目标检测
    • https://github.com/Robert-JunWang/PeleeNet
    • https://github.com/RangiLyu/nanodet
      • https://blog.csdn.net/walletiger/article/details/110733604
  • 遥感图像检测
    • https://github.com/SJTU-Thinklab-Det/DOTA-DOAI

# 图像检索（Image Retrieval）

  • https://github.com/gniknoil/Perfect500K-Beauty-and-Personal-Care-Products-Retrieval-Challenge

# 度量学习（Metric Learning）

  • https://github.com/KevinMusgrave/pytorch-metric-learning
  • https://github.com/PyTorchLightning/metrics
  • https://github.com/enochkan/torch-metrics
  • https://github.com/bnu-wangxun/Deep_Metric
  • https://www.aiuai.cn/aifarm1707.html
  • https://www.aiuai.cn/aifarm1697.html
  • 度量学习的攻击方式
    • https://github.com/layumi/U_turn

# 图像增强（Image Augmentation）

  • https://github.com/albumentations-team/albumentations
  • https://github.com/kornia/kornia
  • https://github.com/YU-Zhiyang/opencv_transforms_torchvision
  • https://github.com/aleju/imgaug
  • https://github.com/qq995431104/Copy-Paste-for-Semantic-Segmentation
  • https://github.com/snu-mllab/Co-Mixup
  • https://github.com/clovaai/CutMix-PyTorch

# 单目标跟踪（Single Object Tracking）

  • https://github.com/STVIR/pysot
  • https://github.com/visionml/pytracking
  • https://github.com/zengwb-lx/Yolov5-Deepsort-Fastreid

# 多目标跟踪（Multi Object Tracking）

  • https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch
  • https://github.com/PeizeSun/TransTrack

# 人脸识别

  • https://github.com/cavalleria/cavaface.pytorch
  • https://github.com/ZhaoJ9014/face.evoLVe.PyTorch
  • https://github.com/JDAI-CV/FaceX-Zoo
  • https://github.com/Tencent/TFace
  • https://github.com/deepinsight/insightface
  • https://github.com/IrvingMeng/MagFace

# 人像抠图

  • https://github.com/lizhengwei1992/Semantic_Human_Matting
  • https://github.com/Shirhe-Lyh/deep_image_matting_pytorch
  • https://github.com/xuebinqin/U-2-Net

# 行人计数

  • https://github.com/zengwb-lx/yolov5-deepsort-pedestrian-counting

# AI部署

  • https://github.com/Tencent/ncnn
  • https://github.com/alibaba/MNN
  • https://github.com/openvinotoolkit/openvino
  • 模型剪枝
    • https://github.com/Sharpiless/Pytorch-Auto-Slim-Tools

# 可解释性

  • https://github.com/pytorch/captum

# 刷算法题

  • https://github.com/chefyuan/algorithm-base

# 模板匹配

  • https://github.com/multi-template-matching/MultiTemplateMatching-Python

# utils

  • pytorch sort

    • https://github.com/teddykoker/torchsort

  • 优化器

    • https://github.com/jettify/pytorch-optimizer
    • https://github.com/lessw2020/Best-Deep-Learning-Optimizers

  • AutoML

    • https://github.com/microsoft/nni

  • 配置框架

    • https://github.com/facebookresearch/hydra

  • 实验记录

    • https://github.com/NVIDIA/runx
    • https://github.com/tqdm/tqdm
    • https://github.com/tensorflow/tensorboard
    • https://github.com/sailist/thexp

  • 训练加速

    • https://github.com/NVIDIA/DALI

  • Summary

    • https://github.com/sksq96/pytorch-summary

  • 各类Trick

    • https://github.com/davidtvs/pytorch-lr-finder
    • https://github.com/lartpang/PyTorchTricks
    • https://github.com/adobe/antialiased-cnns/（抗锯齿化）

  • 可视化

    • https://github.com/jacobgil/pytorch-grad-cam
    • https://github.com/yiskw713/SmoothGradCAMplusplus
    • https://github.com/utkuozbulak/pytorch-cnn-visualizations
    • https://github.com/yizt/Grad-CAM.pytorch

  • 统计参数量

    • https://github.com/Lyken17/pytorch-OpCounter

# 官方仓库

  • https://github.com/HRNet

  • https://github.com/Res2Net/

  • https://github.com/pytorch/fairseq/tree/master/fairseq/optim/lr_scheduler

# 可以参考的文档

图像分类竞赛入门文档

  • https://www.kaggle.com/abhinand05/vision-transformer-vit-tutorial-baseline

分布式训练

  • https://tramac.github.io/2019/03/06/%E5%88%86%E5%B8%83%E5%BC%8F%E8%AE%AD%E7%BB%83-PyTorch/
  • https://www.zybuluo.com/huanghaian/note/1711283
  • https://github.com/tczhangzhi/pytorch-distributed
  • https://github.com/horovod/horovod
  • https://github.com/horovod/horovod/blob/master/docs/pytorch.rst
  • https://zhuanlan.zhihu.com/p/294698838
  • https://zhuanlan.zhihu.com/p/276122469
  • https://github.com/jia-zhuang/pytorch-multi-gpu-training
  • https://github.com/rentainhe/pytorch-distributed-training
  • https://github.com/huggingface/accelerate

新手入门的文档

  • https://bbs.cvmart.net/topics/134

项目配置框架

  • https://zhuanlan.zhihu.com/p/106103204

训练加速

# 竞赛代码库

  • https://github.com/cizhenshi/TianchiGuangdong2019_2th
  • https://github.com/ShuaiBai623/AIC2021-T5-CLV

# 音频处理的库

  • https://github.com/KinWaiCheuk/nnAudio

# 可视化的库

  • https://github.com/DesertsX/dataviz-collections
  • https://github.com/utkuozbulak/pytorch-cnn-visualizations

# 做简历

  • https://sspai.com/post/66053

  • https://github.com/CyC2018/Markdown-Resum

# 公众号文章排版

  • https://github.com/lyricat/wechat-format

# 知识蒸馏

  • https://github.com/zhouzaida/channel-distillation
  • https://github.com/Sharpiless/yolov5-knowledge-distillation
  • https://arxiv.org/abs/1805.06361

# 值得阅读的文章

# 值得观看的分享视频

上次更新: 2021/09/26, 00:09:41
LeetCode 面试算法题卡片-2021-0808

←

最近更新
01
Structured Knowledge Distillation for Semantic Segmentation
 06-03
02
README 美化
 05-20
03
常见 Tricks 代码片段
 05-12
更多文章>
Theme by Vdoing | Copyright © 2021-2023 Muyun99 | MIT License
×