visheratin commited on
Commit
6700676
1 Parent(s): 6010ab2

Add ONNX models

Browse files

The model was converted from PyTorch weights using standard ONNX methods. The pipeline for conversion and checking can be found [here](https://colab.research.google.com/drive/1kvf72PsC4LxSrAqnGvmW_ogFxc3qp_Pg?usp=sharing).

`model.onnx` is an optimized version of the exported model. `model_quant.onnx` is a quantized optimized model.

I checked the difference between the original predictions and the ones from the exported models on a portion of the `scene_parse_150` dataset:

1. `model.onnx` - the absolute difference is 127.364 on a 10x150x128x128 matrix, which looks reasonable. The effective difference (difference in per-pixel class predictions) is 0, which is good.
2. `model_quant.onnx` - the absolute difference is 7876547, which is huge. But the effective difference is only 182, which means that the classes are different for only 182 of 10x128x128 pixels (0.1%).

Files changed (2) hide show
  1. model.onnx +3 -0
  2. model_quant.onnx +3 -0
model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:341333427d74b95391227b7d5935c1854094f7c4b7cc0d144dc6720735dba3bd
3
+ size 15225079
model_quant.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86502e3c75c46b10b48547ccab74a3cab0ea93d487fb6aa2d1399ccc4a26367a
3
+ size 4146422