TropComplique / FaceBoxes-tensorflow

A fast face detector
MIT License
179 stars 66 forks source link

Quantize the model #28

Open LucyLu-LX opened 5 years ago

LucyLu-LX commented 5 years ago

Hi @TropComplique ,

Have you ever tried to quantize the model by Quantization-aware training? If so, how is the result?

I am trying to create_training_graph in model.py at line 72 like this. assert mode == tf.estimator.ModeKeys.TRAIN tf.contrib.quantize.create_training_graph(quant_delay=100000) Since I am not familiar with tf.estimator. I don't quite know where to add tf.contrib.quantize.create_eval_graph() to export a quantized graph before a savedmodel.

Or do I need to rewrite the project without tf.estimator.

If you get any ideas, please let me know.

Thanks a lot.

gray255 commented 4 years ago

Hi, Did you figure this out? I am facing the same issue.