You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I transformed the weights of yolox from torch to tensorflow keras, and now i want to transform keras model to tflite. But when using int8 quantization, the output array of confidence and class ([1, 8500, 5:]) are all 0. I found it's because yolox performed convolution on box, confidence, and class, respectively, in yolo_head.py. And box outputs are not activated with sigmoid, it causes problem for concatenation layers when the input ranges have a great difference. So, how to fix this problem?
reacted with thumbs up emoji reacted with thumbs down emoji reacted with laugh emoji reacted with hooray emoji reacted with confused emoji reacted with heart emoji reacted with rocket emoji reacted with eyes emoji
-
I transformed the weights of yolox from torch to tensorflow keras, and now i want to transform keras model to tflite. But when using int8 quantization, the output array of confidence and class ([1, 8500, 5:]) are all 0. I found it's because yolox performed convolution on box, confidence, and class, respectively, in yolo_head.py. And box outputs are not activated with sigmoid, it causes problem for concatenation layers when the input ranges have a great difference. So, how to fix this problem?
Beta Was this translation helpful? Give feedback.
All reactions