How use a pretrained CNN classifier models transfer to the Faster RCNN to object detection?

endernewton / tf-faster-rcnn

Tensorflow Faster RCNN for Object Detection

https://arxiv.org/pdf/1702.02138.pdf

MIT License

3.65k stars 1.57k forks source link

How use a pretrained CNN classifier models transfer to the Faster RCNN to object detection? #216

Open wmf1991yeah opened 7 years ago

wmf1991yeah commented 7 years ago

I train a CNN classifier model using vgg16 and get a ckpt model files. I want use the pretrained models transfer to the Faster RCNN to object detection. How to do? How to add the ckpt file to your code?

engineer1109 commented 6 years ago

mkdir -p data/imagenet_weights cd data/imagenet_weights wget -v http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz tar -xzvf vgg_16_2016_08_28.tar.gz mv vgg_16.ckpt vgg16.ckpt