cvnet

have fun with image AI

APACHE-2.0 License

Downloads
29
Stars
4
Committers
3

cvnet

Model for Computer Vision(CV) Neural Network.

图像分类

image classification

图像分割

  • 语义分割
  • 实例分割
  • 全景分割

技术演化路径

  1. 2010年前,传统分割:1)边缘检测;2)遗传算法
  2. 2010-2015年,机器学习:1)随机森林;2)支持向量机
  3. 2015年后,深度学习:1)经典分割算法:FCN, U-Net, SegNet, DeepLab; 2)实时分割算法:ENet, LinkNet, BiSeNet, DFANet, Light-Weight RefineNet; 3)RGB-D分割算法:RedNet, RDFNet

MEAL-V2

https://github.com/szq0214/MEAL-V2

介绍他们如何通过蒸馏(distillation)训练一个强大的小模型。所提出方法使用相同模型结构和输入图片大小的前提下, 在 ImageNet 上的性能远超之前 state-of-the-art 的 FixRes 2.5% 以上,甚至超过了魔改结构的 ResNeSt 的结果。

https://www.jiqizhixin.com/articles/2020-09-29-2

Networks implemented

  • PSPNet - With support for loading pretrained models w/o caffe dependency
  • ICNet - With optional batchnorm and pretrained models
  • FRRN - Model A and B
  • FCN - All 1 (FCN32s), 2 (FCN16s) and 3 (FCN8s) stream variants
  • U-Net - With optional deconvolution and batchnorm
  • Link-Net - With multiple resnet backends
  • Segnet - With Unpooling using Maxpool indices

Upcoming

DataLoaders implemented

Demo

  1. demo site: https://www.remove.bg/upload
  2. 演示效果:

demo1:

remove background:

demo2:

remove background:

Reference

  1. ClassyVision
  2. Deep-Learning-Project-Template
  3. pytorch-semseg
  4. torchcv
  5. pytorch-cnn-finetune
  6. PaddleOCR