Skip to content

reproduce the simplest cv work including classification, object detection, segmentation etc.

License

Notifications You must be signed in to change notification settings

pioneer158/SimpleCVReproduction

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

90 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SimpleCVReproduction

一些论文复现,比如Attention 模块。

如果模块过多,则使用多个python模块进行构建,尽量不新建文件夹

将感兴趣/推荐的模型也放在这个库中,以供学习。

尽量提供简化版本的,便于理解的模型文件。

  • Simple_CenterNet 是一个简化版本的,正在试验中。

  • SmallObjectAugmentation是一个专门用于小目标增强库,实际效果不是很理想。增加了一些处理工具模块。

  • attention 实现或者复制官方的pytorch实现,即插即用的注意力模块。

  • captcha-CTC-loss CTC loss+ LSTM

  • deep_sort-master 官方实现,通过该库理解了标准的输入输出格式。

  • easy-receptive-fields-pytorch-master: 用于计算pytorch常用CNN的感受野,非常方便

  • kalman 知乎上的一个简单的卡尔曼滤波算法实现代码

  • opencv-mot 用OpenCV中自带的跟踪器如KCF等实现跟踪,第一帧目标需要在代码中指定。

  • pytorch-commen-code pytorch中常用的一些代码

  • pytorch-grad-cam-master grad cam的实现

  • pytorch-semseg pytorch实现语义分割,目前仅在自己数据集上训练了Unet,无法收敛。

  • siamese-triplet : 孪生网络+triplet loss

  • simple-DCGAN : DCGAN, 还没来得及研究

  • simple-faster-rcnn-pytorch 陈云老师的实现

  • simple-triple-loss 自己仿照一个库写了一个简化版的triple loss

  • tiny_classifier : 目标检测级联一个分类网络中的分类网络的简单实现。

  • tools: 目前只有voc2coco.py工具

  • yolov3-6: U版yolov3中release出来的稳定版本,其中使用的是原始的yolov3 loss,改动不多。

About

reproduce the simplest cv work including classification, object detection, segmentation etc.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 43.2%
  • Jupyter Notebook 41.6%
  • Cuda 5.2%
  • C 3.1%
  • MATLAB 2.5%
  • C++ 1.7%
  • Other 2.7%