Git Product home page Git Product logo

pyramidbox's Introduction

人脸检测模块 仲耀晖

安装环境

  • Ubuntu 16.04 LTS
  • Python3
  • Tensorflow-gpu 1.4
  • OpenCV3

当前模型效果（数据集为 WIDER FACE 的验证集）:

Method AP Easy AP Medium AP Hard
当前模型 90.6 88.8 73.4

运行方法

python demo.py

模型训练方法

  1. 下载 VGG 16 模型 并将其放置于 /checkpoints.
  2. 下载 WIDER FACE 数据集 并转换为 VOC 格式. 路径格式为:
datasets/
       |->widerface/
       |    |->WIDER_train/
       |    |->WIDER_val/
       |    |->WIDER_test/
       |    |->Annotations/
       |    |->JPEGImages/
       |    |...
  1. 使用这个命令来将数据转为 tfrecord 格式的数据:
python datasets/pascalvoc_to_tfrecords.py
  1. 训练策略分为两步: 先将 train_model.py 中的设置改为如下设置再运行，用来训练 PyramidBox 的额外网络层:
self.fine_tune_vgg16 = False
  1. 然后设置 self.fine_tune_vgg16 =Ture 并运行 train_model.py 来训练整个网络.

测试

使用以下脚本进行测试：

python widerface_eval.py
cd eval/eval_tools
octave wider_eval.m

引用

SSD-Tensorflow
SSD_tensorflow_VOC

pyramidbox's People

Contributors

ericzgw avatar huiofficial 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.