search

MaybeShewill-CV / bisenetv2-tensorflow

Unofficial tensorflow implementation of real-time scene image segmentation model "BiSeNet V2: Bilateral Network with Guided Aggregation for Real-time Semantic Segmentation"
https://maybeshewill-cv.github.io/bisenetv2-tensorflow/
MIT License
224 stars 59 forks source link

Quantized Model? #48

Closed fjparrado closed 2 years ago

fjparrado commented 3 years ago

Hi,

I am trying to use the trained model in my TPU, so I need to quantize the model (using TF1).

I think I must add the line contrib_quantize.create_training_graph(input_graph=self._sess.graph, quant_delay=0) inside the __init__ function of the file BiseNetV2CityScapesTrainer.py. However it is a bit unclear to me exactly where .

Would you have any recommendation? Thanks

MaybeShewill-CV commented 3 years ago

@dragonvenenoso Sry doesn't have any experience in tf quant:)

fjparrado commented 3 years ago

Thanks for the answer. Yes... TF1 Quant is a nightmare

MaybeShewill-CV commented 3 years ago

@dragonvenenoso I will keep you informed if I get something valuable:)