Skip to content

use faster RCNN model to detect the passerby from video by tensorflow

License

Notifications You must be signed in to change notification settings

PancakeAwesome/aquatic_products_detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

aquatic_products_detection

This is a object detection module in a environment protection project.We use faster RCNN model to detect the aquatic products under the water so that we can accuqire water qualitiy.Specificly, we firstly receive videos from camera in water and then extract a series of frames from it.Then we can detect these frames by faster RCNN model.The origin paper can be found here. For more detail about the paper and code, see this blog


setup

  • requirements: tensorflow1.3, opencv-python
  • if you have a gpu device, build the library by
cd lib
chmod +x make.sh
./make.sh

demo

  • put your images in data/demo, the results will be saved in data/results, and run demo in the root
python ..//tools/demo.py

training

prepare data

  • First, download the pre-trained model of VGG16 net and put it in data/pretrain/VGG_imagenet.npy.
  • Second, put your own train data and test data in /data,and modify the data path in ./train_net.py
## train 
Simplely run
```shell
python train_net.py
  • you can modify some hyper parameters in ctpn/text.yml, or just used the parameters I set.

About

use faster RCNN model to detect the passerby from video by tensorflow

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages