search

omni-us / squeezedet-keras

Keras implementation of the Squeeze Det Object Detection Deep Learning Framework
MIT License
129 stars 41 forks source link

How to use pretrained model from TF #9

Open cygerts opened 6 years ago

cygerts commented 6 years ago

Hi, how can I use pretrained TF model (on KITTI dataset) from original repo: https://github.com/BichenWuUCB/squeezeDet ? It is stored as binary file there, whereas we need hdf5 file.

Or perhaps you can provide your fine-tuned model on KITTI dataset? (Unfortunately no GPU here no train myself).

Sileadim commented 6 years ago

If you want to convert the original checkpoint, check out this. https://github.com/keras-team/keras/issues/8026 I have not a properly trained model on KITTI, just for testing the implementation. Although, I just pushed a model that was trained for around 250 epochs and you can use that as a starting point.