missing deps

[weberhen]

Thanks for the amazing work!

When trying to use the semantic segmentation part, I miss some libraries, so I had to install them like

pip install einops timm

[]s