Code Monkey home page Code Monkey logo

rfsong-779's Introduction

RFSong-779进行行人检测

重新设计的RFBNet300,模型参数量只有0.99MB,AP达到0.78(相比RFBNet300降了4个点,相比RFong7993降了两个点),速度可以达到200FPS。 RFSong779虽然参数量更小,但是速度与RFSong7993差不多,而精度低了两个点,因此更推荐RFSong7993。后面也利用RFSong7993版本做了很多其他实验,例如钢筋检测、人手检测还有人头检测,表现都还是很好的。

详细讲解请看文章:我的博客

环境

在python3.6 Ubuntu16.04 pytorch1.1下进行了实验

  • 编译NMS

./make.sh

训练自己的数据集

  1. 新建一个文件夹VOCdevkit,将自己的VOC格式数据集的VOC2007文件夹移动到VOCdevkit中,例如我的就是/home/common/wangsong/VOCdevkit/VOC2007
$VOCdevkit/
$VOCdevkit/VOC2007/
$VOCdevkit/VOC2007/Annotations/
$VOCdevkit/VOC2007/ImageSets/
$VOCdevkit/VOC2007/JPEGImages/
  1. 修改data/config.py中的数据集路径 VOCroot = '/home/common/wangsong/VOC/VOCdevkit' 改为自己数据集对应的路径

  2. 修改针对VOC行人的代码部分,可以参考我的博客查看修改的地方

  • train_RFB.py的trainsets进行改动:train_sets = [('2007', 'trainval'))]
  • data/voc0712.py和data/voc_eval.py这两个文件可以直接用钢筋检测里面的替换
  1. 运行train_RFB

与RFB不同的是

  • 代码更方便进行自己设计网络
  • 网络非常轻量级只有0.99 MB
  • 网络速度相比RFB有显著提升,能够达到200FPS

欢迎加群交流,云深不知处-目标检测 763679865

rfsong-779's People

Contributors

songwsx avatar

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.