102-pytorch-onnx-to-openvino

PyTorch to ONNX and OpenVINO™ IR Tutorial

This notebook demonstrates how to do inference on a PyTorch semantic segmentation model, using OpenVINO.

Notebook Contents

The notebook uses Model Optimizer to convert the open source fastseg semantic segmentation model, trained on the CityScapes dataset, to OpenVINO IR. It also shows how to do segmentation inference on an image, using OpenVINO Runtime and compares the results of the PyTorch model with the OpenVINO IR model.

Installation Instructions

If you have not installed all required dependencies, follow the Installation Guide.