[Build flows] Update download scripts for updated models

build/cybersecurity-mlp/build.py

@@ -31,7 +31,6 @@
 from finn.util.basic import alveo_default_platform
 import os
 import shutil
-from custom_steps import custom_step_mlp_export
 
 # Which platforms to build the networks for
 zynq_platforms = ["Pynq-Z1", "Ultra96", "ZCU104"]
@@ -90,8 +89,7 @@ def platform_to_shell(platform):
         save_intermediate_models=True,
     )
 
-    # Export MLP model to FINN-ONNX
-    model = custom_step_mlp_export(model_name)
+    model = "models/%s.onnx" % model_name
     # Launch FINN compiler to generate bitfile
     build.build_dataflow_cfg(model, cfg)
     # Copy bitfiles into release dir if found

build/cybersecurity-mlp/models/download-model.sh

@@ -0,0 +1,4 @@
+#!/bin/sh
+
+wget https://github.com/Xilinx/finn-examples/releases/download/v0.0.7a/onnx-models-cybersecurity.zip
+unzip -j onnx-models-cybersecurity.zip

build/gtsrb/models/download-model.sh

@@ -1,2 +1,3 @@
 #!/bin/bash
-wget https://github.com/fastmachinelearning/qonnx_model_zoo/raw/main/models/GTSRB/Brevitas_CNV1W1A/cnv_1w1a_gtsrb.onnx
+wget https://github.com/Xilinx/finn-examples/releases/download/v0.0.7a/onnx-models-gtsrb.zip
+unzip -j onnx-models-gtsrb.zip

build/kws/build.py

@@ -67,7 +67,7 @@ def step_preprocess(model: ModelWrapper, cfg: DataflowBuildConfig):
     build_cfg.VerificationStepType.FOLDED_HLS_CPPSIM,
 ]
 
-model_name = "MLP_W3A3_python_speech_features_pre-processing_QONNX"
+model_name = "MLP_W3A3_python_speech_features_pre-processing_QONNX_opset-11"
 model_file = "models/" + model_name + ".onnx"
 
 # Change the ONNX opset from version 9 to 11, which adds support for the TopK node

build/kws/models/download-model.sh

@@ -29,4 +29,5 @@
 
 # Download validation data and model
 wget https://github.com/Xilinx/finn-examples/releases/download/kws/python_speech_preprocessing_all_validation_KWS_data.npz
-wget https://github.com/Xilinx/finn-examples/releases/download/kws/MLP_W3A3_python_speech_features_pre-processing_QONNX.onnx
+wget https://github.com/Xilinx/finn-examples/releases/download/v0.0.7a/onnx-models-kws.zip
+unzip -j onnx-models-kws.zip

build/mobilenet-v1/build.py

@@ -148,7 +148,7 @@ def select_build_steps(platform):
         specialize_layers_config_file="specialize_layers_config/%s_specialize_layers.json"
         % platform_name,
     )
-    model_file = "models/%s_pre_post_tidy.onnx" % model_name
+    model_file = "models/%s_pre_post_tidy_opset-11.onnx" % model_name
     build.build_dataflow_cfg(model_file, cfg)
 
     # copy bitfiles and runtime weights into release dir if found

build/mobilenet-v1/models/download-model.sh

@@ -1,4 +1,4 @@
 #!/bin/sh
 
-wget https://github.com/Xilinx/finn-examples/releases/download/v0.0.1a/onnx-models-mobilenetv1.zip
-unzip onnx-models-mobilenetv1.zip
+wget https://github.com/Xilinx/finn-examples/releases/download/v0.0.7a/onnx-models-mobilenetv1.zip
+unzip -j onnx-models-mobilenetv1.zip

build/resnet50/models/download-model.sh

@@ -1,4 +1,4 @@
 #!/bin/sh
 
-wget https://github.com/Xilinx/finn-examples/releases/download/v0.0.1a/onnx-models-resnet50.zip
-unzip onnx-models-resnet50.zip
+wget https://github.com/Xilinx/finn-examples/releases/download/v0.0.7a/onnx-models-resnet50.zip
+unzip -j onnx-models-resnet50.zip

build/vgg10-radioml/models/download-model.sh

@@ -1,3 +1,4 @@
 #!/bin/sh
 
-wget https://github.com/Xilinx/finn-examples/releases/download/radioml/radioml_w4a4_small_tidy.onnx
+wget https://github.com/Xilinx/finn-examples/releases/download/v0.0.7a/onnx-models-radioml.zip
+unzip -j onnx-models-radioml.zip