/SimpleCVReproduction

Replication of simple CV Projects including attention, classification, detection, keypoint detection, etc.

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

SimpleCVReproduction

Recommended models are listed in this repository. In order to simplify learning for beginners, comments are provided with running models, code ready to read, well-documented code, and a series of simple code snippets. The aim of this project is to provide a simplified version of easy-to-understand model files. Suggestions for easily learnable libraries are welcome within the Issues section. Most of the content in this project is from Github and shall not be used for commercial purposes. In case of any infringement, please contact the author for removal.

将感兴趣/推荐的模型也放在这个库中,以供学习。由于好多库从头开始学习难度太大,在这里提供了笔者的部分注释,其中大部分都是跑过的模型、准备读的代码、已经读过的代码笔记、以及开发的simple系列简单代码、常用代码段等。

本项目致力于提供简化版本的,便于理解的模型文件。

如果有推荐的便于初学者学习的库,也欢迎在issue中提出和补充。

本项目大部分内容是来源于Github,不会用做商业用途,如有侵权,请联系笔者删除。

目录

即插即用模块&注意力模块

原项目已经迁移至新的地址:Awesome-Attention-Mechanism-in-cv

主要内容包括:

  • 计算机视觉领域中注意力模块。
  • 计算机视觉中即插即用模块。code
  • Vision Transformer系列工作。

更多介绍:

项目推荐

项目 介绍 链接
CenterNet 简化版本的CenterNet目标检测算法(第三方实现) link
SmallObjectAugmentation 针对小目标进行数据增强库,在笔者数据集效果不理想 link
DarkLabel 专门用于DarkLabel软件转化的系列脚本 link
Latex/latex_algo 用latex写的伪代码示例 link
MLP MLP-Mixer,ResMLP,RepMLP简单源码 link
NAS 感兴趣的神经网络结构搜索算法 link
Plug-and-play Module 即插即用模块 link
52RL 参加DataWhale深度强化学习课程代码 code link
Vision Transformer 最经典的ViT实现, 训练代码在code link
captcha-CTC-loss CTC loss+ LSTM link
cifarTrick 原先收集的部分Trick更多Trick在Tricks link
deep_sort 官方实现的DeepSort算法 link
deep_sort_yolov3_pytorch 笔者自己实现和改进的DeepSort算法 link
easy-receptive-fields 感受野计算,分析,特征图可视化 link
fine_grained_baseline 细粒度识别baseline,Bilinear Pooling操作 link
flask-yolo flask配合yolo算法实现网页 link
kalman 卡尔曼滤波实现与测试 link
libfacedetection.train 人脸检测训练代码 link
opencv-mot 使用Opencv实现多目标跟踪 link
pandoc-starter Pandoc是Markdown转化器,很方便 link
pytorch-commen-code 常用的pytorch代码片段 link
siamese-triplet 孪生网络+Triplet Loss实现 link
simple-triple-loss 笔者自己实现的triplet loss link
simple_keypoint [推荐] 笔者极简代码实现关键点识别,提供根据heatmap进行识别的方法 link
tikz_cnn 使用latex绘制CNN图 link
tsne tsne可视化数据集 link
tools voc2coco脚本,yolo anchor聚类脚本 link
tiny_classifier 超级简单的分类代码+focal loss使用 link
yolov3-6 第六次release版本,属于老版本yolo实现 link

致谢

@zhongqiu1245 补充的borderDet中的BAM模块,补充了FPT

@1187697147 补充的context-gating模块

@cmsfw-github 指出了simple_keypoint中的bug

@1187697147 建议更新了AFF和iAFF模块源码

贡献

欢迎在issue中提出补充推荐的项目。

欢迎关注“GiantPandaCV”公众号以及“神经网络架构搜索”公众号查看相关博客。