xiaohongxiao / thundernet-tensorflow2.0

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

purpose

please refer to https://github.com/mohhao/TF-Keras-ThunderNet

this project is constructed for myself and anyone else who need it to learn tensorflow2.0 and not intend to reproducing the original result of the paper. API inside tf.keras.Backend is prefered than tf.

install kerasflow:

pip install tensorflow-gpu==2.0.0-beta1

To Do

  • implement backbone model
  • update light-head-rcnn-tf to tensorflow 2.0
  • integrate detection head
  • Implement data input pipeline using tf.data.Dataset
  • Train on voc, coco

About

License:MIT License


Languages

Language:Python 100.0%