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.