[Gen2] Creating custom models

This experiment can be used as a tutorial on how users can create custom models with PyTorch/Kornia, convert them and run them on the DepthAI.

Tutorial documentation here.

blur.py, concat.py and edge.py are scripts that run created custom models. generate_model/ folder contains scripts that create these custom models (frame blurring, frame concatenation and edge detection).

Demos

Concatenate frames

Blur frames

Corner detection

Difference between 2 frames

diff.py computes the difference between two consecutive color frames and visualizes that along with the color frame.

Pre-requisites

Install requirements

python3 -m pip install -r requirements.txt