search

BichenWuUCB / squeezeDet

A tensorflow implementation for SqueezeDet, a convolutional neural network for object detection.
BSD 2-Clause "Simplified" License
739 stars 306 forks source link

How to convert checkpoint of squeezedet to frozen graph for tflite conversion?! #124

Open kiad4631 opened 5 years ago

kiad4631 commented 5 years ago

I try to convert the squeezedet checkpoint to frozen graph and then conversion tflite but I cant. my command is:

python3 freeze_graph.py --input_meta_graph=/home/davari/detection_on_mobile/squeezeDet/model.ckpt-95000.meta --input_checkpoint=/home/davari/detection_on_mobile/squeezeDet/model.ckpt-95000 --output_node_names="bbox/trimming/bbox,probability/class_idx,probability/score" --input_binary=True --output_graph=/home/davari/detection_on_mobile/squeezeDet/model-95000.pb

is it correct conversion?!

bayesian-mind commented 4 years ago

Did you find this out, I am trying to do the same thing?