R-FCN Tensorflow version for primary students
Mainly refer to https://github.com/auroua/tf_rfcn and https://github.com/endernewton/tf-faster-rcnn
Thanks for auroua and endernewton's help.
This code is for primary students to use.
What is different from theirs?
-
Support Python3 Train and Test
-
Support Chinese Language of detecting result
-
More easy to use and small changes
System: Unix or Unix-Like Platrform
(Have only tested on linux-ubuntu, other unix system should be supported in theory)
Python3
Tensorflow 1.8+
OpenCV-Python
CUDA
CUDNN
You need to download weights first. See the following Models Download. Inference
git clone https://github.com/engineer1109/TensorFlow-R-FCN-for-primary-students.git
cd lib && make
cd ..
sh run.sh
Train
sh train.sh
Tensorboard viewer
sh tensorboard.sh
Pascal Voc 2007+2012
Unzip it to ./data/VOCdevkit2007/VOC2007
Main Model Unzip it to ./output
Download | Path |
---|---|
Baidu Yun | https://pan.baidu.com/s/1-zAgNRavUJGX5ggmSH-TEQ Code: b4vt |
https://drive.google.com/open?id=1iWMqZAX3wh-XN0o5pIGClvIx8wMPWWFT |
Pretrained Model Unzip it to ./data/imagenet_weights
Download | Path |
---|---|
Baidu Yun | https://pan.baidu.com/s/1oSZMlwOaIFOuqrMeSIY-PQ Code: qvos |
wget | http://download.tensorflow.org/models/resnet_v1_101_2016_08_28.tar.gz |
For wget, you need to change filename to res101.ckpt
wget -v http://download.tensorflow.org/models/resnet_v1_101_2016_08_28.tar.gz
tar -xzvf resnet_v1_101_2016_08_28.tar.gz
mv resnet_v1_101.ckpt res101.ckpt