sony / mct_quantizers

Apache License 2.0
17 stars 8 forks source link

Add custom onnx ops to LUT quantizers + Fix Pytorch LUT apis #57

Closed reuvenperetz closed 1 year ago

reuvenperetz commented 1 year ago

Add pytorch LUT quantizers an option to be exported in ONNX. Change LUT quantizers API to receive lists instead of np arrays (for lut values and thresholds). Adapt all LUT tests that pass np arrays to lists. Fix the issue with ignoring the channel axis in LUT quantizers (and tests that assumed the channel axis is always 0)