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”公众号以及“神经网络架构搜索”公众号查看相关博客。