| "A script to generate all development files necessary for the image filtering demo." | |
| import shutil | |
| import onnx | |
| from common import AVAILABLE_FILTERS, FILTERS_PATH | |
| from filters import Filter | |
| from client_server_interface import FHEDev | |
| print("Generating deployment files for all available filters") | |
| for filter_name in AVAILABLE_FILTERS: | |
| print("Filter:", filter_name, "\n") | |
| # Create the filter instance | |
| filter = Filter(filter_name) | |
| # Compile the model on a representative inputset | |
| filter.compile() | |
| # Define the directory path associated to this filter | |
| filter_path = FILTERS_PATH / filter_name | |
| # Define the directory path associated to this filter's deployment files | |
| deployment_path = filter_path / "deployment" | |
| # Delete the deployment folder and its content if it already exists | |
| if deployment_path.is_dir(): | |
| shutil.rmtree(deployment_path) | |
| # Save the files needed for deployment | |
| fhe_dev_filter = FHEDev(filter, deployment_path) | |
| fhe_dev_filter.save() | |
| # Save the ONNX model | |
| onnx.save(filter.onnx_graph, filter_path / "server.onnx") | |
| print("Done !") | |