/tf-pvanet

Implementation of PVANet in Tensorflow

Primary LanguagePython

tf-pvanet

Implementation of PVANet in Tensorflow.

  • When repeating the network structure in caffe pt, the original layer names are carefully reserved.
  • The caffe_to_tensorflow.py converts pva model in caffe to tensorflow ckpt according to the layer names. Caffe models can be downloaded via this script.
  • Only the conv stages from conv1_1 to conv5_4 are implemented, as an base network archietecture for any other CNN-based algorithms, more than faster-rcnn.
  • The util module come from pylib. Add the path of its src directory to PYTHONPATH when using.