1!pip install --upgrade onnx onnxsim
2import onnx
3from onnxsim import simplify
4from onnx import version_converter
5import urllib.request
6MODEL_URL = "https://github.com/Linzaer/Ultra-Light-Fast-Generic-Face-Detector-1MB/raw/master/models/onnx/version-RFB-640.onnx"
7INPUT_FILENAME = "rfb640_original.onnx"
8urllib.request.urlretrieve(MODEL_URL, INPUT_FILENAME)
9model = onnx.load(INPUT_FILENAME)
10optimized_model = version_converter.convert_version(model, 17)
11model_simp, check = simplify(
12 optimized_model,
13 overwrite_input_shapes={'input': [1, 3, 480, 640]}
14)
15onnx.save(model_simp, "version-RFB-640_simplified_fixed.onnx")
