junohang / SimpleCVReproduction

:punch: Reproduce simple cv project including attention module, classification, object detection, segmentation, keypoint detection, tracking :smile: etc.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SimpleCVReproduction

将感兴趣/推荐的模型也放在这个库中,以供学习。由于好多库从头开始学习难度太大,在这里提供了笔者的部分注释,其中大部分都是跑过的模型、准备读的代码、已经读过的代码笔记、以及开发的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”公众号以及“神经网络架构搜索”公众号查看笔者发布的第一手博客。

About

:punch: Reproduce simple cv project including attention module, classification, object detection, segmentation, keypoint detection, tracking :smile: etc.

License:Apache License 2.0


Languages

Language:Python 64.0%Language:Jupyter Notebook 29.6%Language:TeX 1.7%Language:Cuda 1.2%Language:Shell 0.9%Language:C 0.7%Language:C++ 0.6%Language:MATLAB 0.6%Language:Cython 0.2%Language:Lua 0.2%Language:Makefile 0.2%Language:CMake 0.1%Language:Dockerfile 0.0%Language:Batchfile 0.0%Language:PowerShell 0.0%Language:CSS 0.0%