将感兴趣/推荐的模型也放在这个库中,以供学习。由于好多库从头开始学习难度太大,在这里提供了笔者的部分注释,其中大部分都是跑过的模型、准备读的代码、已经读过的代码笔记、以及开发的simple系列简单代码、常用代码段等。
本项目致力于提供简化版本的,便于理解的模型文件。
如果有推荐的便于初学者学习的库,也欢迎在issue中提出和补充。
本项目大部分内容是来源于Github,不会用做商业用途,如有侵权,请联系笔者删除。
原项目已经迁移至新的地址:Awesome-Attention-Mechanism-in-cv
主要内容包括:
- 计算机视觉领域中注意力模块。
- 计算机视觉中即插即用模块。code
- Vision Transformer系列工作。
更多介绍:
项目 | 介绍 | 链接 |
---|---|---|
CenterNet | 简化版本的CenterNet目标检测算法(第三方实现) | link |
SmallObjectAugmentation | 针对小目标进行数据增强库,在笔者数据集效果不理想 | link |
DBFace | 实时单阶段人脸检测器 | link |
DarkLabel | 专门用于DarkLabel软件转化的系列脚本 | link |
Latex/latex_algo | 用latex写的伪代码示例 | link |
MLP | MLP-Mixer,ResMLP,RepMLP简单源码 | link |
NAS | 感兴趣的神经网络结构搜索算法 | link |
Plug-and-play Module | 即插即用模块 | link |
PyTorch-Lightning | Lightning的基础使用案例 | link |
52RL | 参加DataWhale深度强化学习课程代码 code | link |
Vision Transformer | 最经典的ViT实现, 训练代码在code | link |
YOLOv3-complete-pruning | YOLOv3经典的剪枝算法 | link |
captcha-CTC-loss | CTC loss+ LSTM | link |
cifarTrick | 原先收集的部分Trick更多Trick在Tricks | link |
cvtransforms | 数据增强方法, 可以替代pytorch中transform(PIL-based),据说让数据读取快三倍 | link |
deep_sort | 官方实现的DeepSort算法 | link |
deep_sort_yolov3_pytorch | 笔者自己实现和改进的DeepSort算法 | link |
easy-receptive-fields-pytorch | 感受野计算 | 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 |
pytorch-grad-cam | Grad Cam代码实现,特征图可视化 | link |
pytorch-semseg | 语义分割代码库收集,经测试无法收敛(私有数据集) | link |
siamese-triplet | 孪生网络+Triplet Loss实现 | link |
simple-faster-rcnn-pytorch | 陈云老师的faster rcnn实现 | 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”公众号以及“神经网络架构搜索”公众号查看相关博客。