Giter Club home page Giter Club logo

pprp / simplecvreproduction Goto Github PK

View Code? Open in Web Editor NEW
1.2K 23.0 310.0 259.32 MB

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

License: Apache License 2.0

Python 44.42% Jupyter Notebook 47.58% Shell 0.71% Cuda 1.74% C 1.03% C++ 0.57% Makefile 0.20% TeX 2.31% Lua 0.30% MATLAB 0.83% CSS 0.01% Batchfile 0.01% PowerShell 0.01% Cython 0.31%
computer-vision paper-reproduction demo cv pytorch classification object-detection face-detection landmark attention

simplecvreproduction's Introduction

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

simplecvreproduction's People

Contributors

pprp avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

simplecvreproduction's Issues

integrate with Lightning ecosystem CI

Hello and so happy to see you use Pytorch-Lightning! 🎉
Just wondering if you already heard about quite the new Pytorch Lightning (PL) ecosystem CI where we would like to invite you to... You can check out our blog post about it: Stay Ahead of Breaking Changes with the New Lightning Ecosystem CI
As you use PL framework for your cool project, we would like to enhance your experience and offer you safe updates to our future releases. At this moment, you run tests with a particular PL version, but it may accidentally happen that the next version will be incompatible with your project... 😕 We do not intend to change anything on our project side, but still here we have a solution - ecosystem CI with testing both - your and our latest development head we can find it very early and prevent releasing eventually bad version... 👍

What is needed to do?

What will you get?

  • scheduled nightly testing configured for development/stable versions
  • slack notification if something went wrong to investigate
  • testing also on multi-GPU machine as our gift to you 🐰

cc: @Borda

Darklabel 转 MOT

Darklabel 转 MOT ,请问一下 det.txt 中的 conf 置信度和 “遮挡” 参数怎么获取呢?

camvid data structure

dear author,

can you please tell me the dataset structure of camvid dataset.

i split images into train val and test. where should be the labels/ground truth image?

关于pascal.py里的annopath

作者您好,当我运行到这句代码时self.annopath = os.path.join(VOC_test_root, 'VOC2007', 'Annotations', '{:s}.xml')recs[imagename] = self.parse_record(annopath.format(imagename))总是报错找不到.xml文件,显示路径有错误,不知道您有没有遇到类似的问题。`

Sematic Embbed Block(SEB)

作者你好,我看你的知乎博文,里面提到了SEB模块,关于这个模块,这个github的代码好像并没有添加,请问作者是在哪里看到的这个SEB模块呢?

bug 发现

utils的这个函数,shape定义错了,应当是 h,w = heatmap.shape
原来的是w,h
def draw_umich_gaussian(heatmap, center, radius, k=1):
diameter = 2 * radius + 1
gaussian = gaussian2D((diameter, diameter), sigma=diameter / 6)
# print(gaussian.shape)
# 一个圆对应内切正方形的高斯分布

x, y = int(center[0]), int(center[1])

height, width = heatmap.shape

left, right = min(x, radius), min(width - x, radius + 1)
top, bottom = min(y, radius), min(height - y, radius + 1)

masked_heatmap = heatmap[y - top:y + bottom, x - left:x + right]
masked_gaussian = gaussian[radius - top:radius +
                           bottom, radius - left:radius + right]

if min(masked_gaussian.shape) > 0 and min(masked_heatmap.shape) > 0:  # TODO debug
    np.maximum(masked_heatmap, masked_gaussian * k, out=masked_heatmap)
    # 将高斯分布覆盖到heatmap上,取最大,而不是叠加
return heatmap

gpu utilization

hello author,

i am using your pytorch segmentation implementation. the gpu is not being used. the system is using only cpu. i have cuda device in pytorch. also, why is batch size 1 in your config and what n_workers should i use?

thank you

mot evaluate

能不能直接用cvat或者darklabel标注视频输出gt,再与deep sort输出的result进行比较输出mota、ids这些指标

Multi-object in Infrared small target detection

哈喽,

请问作者有没有试过heatmap方法检测多个红外目标中心点?
如果有试过的话效果如何?
还有就是,如果我想尝试这个任务的话,只需要修改dataloader应该就可以了吧?
@pprp

多谢!

from tensorboardX import SummaryWriter

hello author,

i am interested in your pytorch segmentation implementation. i faced some problem while training the bash file as follows,

File "train.py", line 27, in
from tensorboardX import SummaryWriter
File "/home/rahul/anaconda3/envs/semseg/lib/python2.7/site-packages/tensorboardX/init.py", line 5, in
from .torchvis import TorchVis
File "/home/rahul/anaconda3/envs/semseg/lib/python2.7/site-packages/tensorboardX/torchvis.py", line 11, in
from .writer import SummaryWriter
File "/home/rahul/anaconda3/envs/semseg/lib/python2.7/site-packages/tensorboardX/writer.py", line 223
logdir: Optional[str] = None,
^
SyntaxError: invalid syntax

thank you

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.